[DK]Programmers - 문자열 내 마음대로 정렬하기 - C++

 

출처 : https://programmers.co.kr/learn/courses/30/lessons/12915?language=cpp 

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

 

algorithm 라이브러리에 sort 함수 사용해서 sort에 기준이되는 함수를 작성했다.

 

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

using namespace std;
int gn;

bool cmp(const string &a, const string &b) {
    return a[gn] == b[gn] ? a < b : a[gn] < b[gn];
}

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