[프로그래머스/C++] ㅡ LV1 - 문자열 내 마음대로 정렬하기

문제

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

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

풀이

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

using namespace std;

int N;
bool cmp (string a, string b) {
        if (a[N] < b[N]) return true;
        else if (a[N] > b[N]) return false; // n번째 문자열 비교
        
        if (a < b) return true;	// 전체 문자열 비교
        else return false;
}

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

풀이

cmp 내부에서 if문 두개를 사용했다.

비교할 n번째 문자열이 같다면 1번째 if문은 그냥 통과하게 된다.

그 경우 문자열을 사전순으로 정렬시키면 된다.

 

cmp내부에 n을 넣는 방식이 어려웠었는데, 생각보다 단순한 방식으로 해결이 된다.

cmp 함수 정의 이전에 정의한 변수에 값을 대입시키면 된다.