개발 로그/알고리즘

프로그래머스) 문자열 내 마음대로 정렬하기 c++

CyberSoak 2021. 9. 4. 12:20

string이 vector와 유사함을 착안.

bool compare (string a, string b) {
    return a[i] == b[i] ? a < b : a[i] < b[i];
}

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