[C++] 문자열 내 마음대로 정렬하기 (sort 커스텀)

 

문자열 내 마음대로 정렬하기

문자열로 구성된 리스트 strings와, 정수 n이 주어졌을 때, 각 문자열의 인덱스 n번째 글자를 기준으로 오름차순 정렬하려 합니다. 예를 들어 strings가 ["sun", "bed", "car"]이고 n이 1이면 각 단어의 인덱스 1의 문자 "u", "e", "a"로 strings를 정렬합니다.

 

 

입출력 예

["sun", "bed", "car"]	1	["car", "bed", "sun"]
["abce", "abcd", "cdx"]	2	["abcd", "abce", "cdx"]

 

 

처음에 map으로 주어진 배열의 문자열과 인덱스를 저장하고 정렬하는 코드를 작성하려 했는데, sort 함수를 커스텀하여 그냥 주어진 문자열의 해당 인덱스 값을 기준으로 정렬하는 코드를 작성하여 풀이했습니다.

 

 

풀이

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

using namespace std;

int num = 0;

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

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

 

 

 

 

참고자료

C++ STL sort() 함수 다루기

C++ Map을 value 기준으로 정렬하기

[C++] std::map을 value 기준으로 정렬하기 - 널두