프로그래머스 / 연습문제 / 문자열 내 마음대로 정렬하기

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

프로그래머스

#include <string>
#include <vector>
#include <algorithm>
 
using namespace std;
 
int k;
bool cmp(string a, string b) {
    if (a[k] != b[k]) return a[k] < b[k]; //그대로 정렬
    else return a < b; // 문자가 같을 경우 사전 순 정렬
}
 
vector<string> solution(vector<string> strings, int n) {
    k = n;
    sort(strings.begin(), strings.end(), cmp);
 
    return strings;
    //return answer;
}

sort의 정렬 기준을 따로 설정할 수 있다는 것을 몰라서

생각보다 오래 걸렸던 것 같다.

sort함수의 3번째 인자로 정렬 기준을 잡을 수 있다.