Programmers / 문자열 내 마음대로 정렬하기 / C++

*문제 출처는 프로그래머스에 있습니다.

 

문제 제목: 문자열 내 마음대로 정렬하기 (1단계)

문제 사이트: https://school.programmers.co.kr/learn/courses/30/lessons/120817

프로그래머스

 

문제 설명

 

---

 

나의 풀이

 

#include <string>
#include <vector>
#include <algorithm>
using namespace std;

int m;
bool cmp(string a, string b) {
    if (a[m] == b[m])
        return a < b;
    else 
        return a[m] < b[m];
}

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

    m = n;
    sort(answer.begin(), answer.end(), cmp);

    return answer;
}

---

※ 알아야 할 것

 

sort에서 함수를 넣어서 두 인자를 비교할 수 있다.

신기하네