[C/C++] programmers Level 1 문자열 내 마음대로 정렬하기

 

 

 

 

 

코드

 

#include <string>
#include <vector>
#include <algorithm> // sort()

using namespace std;

int num;

bool compare(string prev, string next) { // 정렬되는 순서는 true가 되는 규칙을 따른다.
    if (prev[num] == next[num]){ //
        return prev < next; //  참이면 1, 거짓이면 0 반환 -> 참일시 sort 실시 오름차순 정렬
    }
    
    return prev[num] < next[num];  // 창이면 1, 거짓이면 0 -> 참일시 sort 실시 오름차순 정렬
}

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