sort()에 조건달아 정렬하기

• compare()에 변수가 필요할 땐 전역으로 선언 후 사용하게 하면 된다!

코드

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

using namespace std;
int index;

bool compare(string a, string b) {
    if (a[index] < b[index]) return true;
    if (a[index] > b[index]) return false;
    return a < b;
}

vector<string> solution(vector<string> strings, int n) {
    vector<string> answer;
    index = n;

    sort(strings.begin(), strings.end(), compare);
    answer = strings;
    return answer;
}