[프로그래머스] 문자열 내 마음대로 정렬하기 (C++)

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

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

문제 설명

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

 

제한 조건

• strings는 길이 1 이상, 50이하인 배열입니다.
• strings의 원소는 소문자 알파벳으로 이루어져 있습니다.
• strings의 원소는 길이 1 이상, 100이하인 문자열입니다.
• 모든 strings의 원소의 길이는 n보다 큽니다.
• 인덱스 1의 문자가 같은 문자열이 여럿 일 경우, 사전순으로 앞선 문자열이 앞쪽에 위치합니다.

입출력 예

strings	n	return
[sun, bed, car]	1	[car, bed, sun]
[abce, abcd, cdx]	2	[abcd, abce, cdx]

입출력 예 설명

입출력 예 1
sun, bed, car의 1번째 인덱스 값은 각각 u, e, a 입니다. 이를 기준으로 strings를 정렬하면 [car, bed, sun] 입니다.

입출력 예 2
abce와 abcd, cdx의 2번째 인덱스 값은 c, c, x입니다. 따라서 정렬 후에는 cdx가 가장 뒤에 위치합니다. abce와 abcd는 사전순으로 정렬하면 abcd가 우선하므로, 답은 [abcd, abce, cdx] 입니다.

 

풀이1

C++ STL sort 함수를 사용해서 정렬했다.

sort함수의 3번째 parameter에 사용자 정의 비교함수명(compare)을 넣어 사용자가 정한 규칙대로 정렬되게 했다.

문제의 n을 compare 함수의 파라미터로 가져올 방법을 찾지 못해 전역변수 nn에 n을 넣어주어 사용했다.

 

코드1

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

int nn;
bool compare(string a, string b){
    if(a[nn]<b[nn]) return true;
    else if (a[nn]==b[nn]){
        return a<b;
    }
    else return false;
}

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

 

풀이2

풀이1에서 해결하지 못한 부분을 람다 함수를 사용함으로써 n을 참조할 수 있도록 했다.

풀이1에 비해 코드가 간결해졌다.

 

코드2

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

vector<string> solution(vector<string> strings, int n) {
    vector<string> answer;
    sort(strings.begin(), strings.end(), [=](string a, string b)->bool {
        if(a[n]<b[n]) return true;
        else if (a[n]==b[n]){
            return a<b;
        }
        else return false;
    });
    answer.assign(strings.begin(),strings.end());
    return answer;
}