[Level 1][C++] 문자열 내 마음대로 정렬하기

#include <string>
#include <vector>
#include <algorithm>

using namespace std;

int idx;

bool compare(string a, string b)
{
    if(a[idx] != b[idx]) return a[idx]<b[idx];  
    return a<b;                                                     //같지 않다면 사전편찬순으로 정렬
}

vector<string> solution(vector<string> strings, int n) {

    idx = n;      
    sort(strings.begin(), strings.end(), compare);    //정렬 함수.  유저가 정의한 함수(compare)대로 정렬한다.

return strings;
}