[프로그래머스/Level 1] 문자열 내 마음대로 정렬하기 (C++)

programmers.co.kr/learn/courses/30/lessons/12915

코딩테스트 연습 - 문자열 내 마음대로 정렬하기

 

풀이

풀이가 필요하진 않지만, 문자열이랑 정렬 문제에 약해서 기록하는 용도!

 

전체 코드

#include <string>
#include <vector>
#include <algorithm>
using namespace std;

int idx;

int cmp(string a, string b) {
    if(a[idx]!=b[idx]) {
        return a[idx]<b[idx];
    } else {
        return a<b;
    }
}

vector<string> solution(vector<string> strings, int n) {
    idx=n;
    sort(strings.begin(),strings.end(),cmp);
    return strings;
}