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

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

프로그래머스

 

문제에 제시된 제한 사항 

입출력 예시

#include <string>
#include <vector>
#include <algorithm>
using namespace std;
 
int N;
 
bool compare_c(string a, string b) {
    if (a.at(N) != b.at(N)) return a.at(N) < b.at(N);
    else return a < b;
}
 
vector<string> solution(vector<string> strings, int n) {
    vector<string> answer;
 
    N = n;
 
    sort(strings.begin(), strings.end(), compare_c);
    answer = strings;
 
    return answer;
}

algorithm 라이브러리의 sort() 함수를 활용해 풀 수 있었다.

sort() 다양한 사용법 중 3번째 인자로 사용자가 정의한 비교함수를 넣어 사용할 수 있는 것을 알면 쉽게 풀 수 있음

n번째 인덱스 원소의 사전적 순서로 비교하는 사용자 정의 함수 compare_c 를 정의하여 sort()에 넣어주었다

 

sort() → algorithm 라이브러리에 정의된 함수로 기본적으로 Quick sort 알고리즘을 사용한다

  1. template <class RandomIt>

     sort(RandomIt begin, RandomIt end)

     → begin에서부터 end까지의 반복자 범위 내 원소들을 오름차순 정렬해준다.

         기본적으로 operator< 를 이용해 원소들을 비교

 

  2. template <class RandomIt, class Compare>

     sort(RandomIt begin, RandomIt end, Compare cmp)

     → begin에서부터 end까지의 반복자 범위 내 원소들을 오름차순 정렬해준다.

         cmp 비교함수를 사용해 비교하여 원소들을 비교 (사용자 정의 함수도 사용 가능)

         비교함수의 반환값은 boolean 타입으로 true일 경우 앞의 원소가 먼저 정렬된다

 

  * sort()의 사용법은 몇 개 더 있지만 두 가지만 정리해 두었습니다.