[프로그래머스] 연습문제 - 문자열 내 마음대로 정렬하기, C++

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

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

 본 문제는 sort와 cmp를 연습하는 문제다. 프로그래머스 LEVEL 1, 2는 C++ STL을 연습할 수 있는 문제가 많아서 좋다. sort 사용에 대해 잠깐 소개하면

• 오름차순 - sort(v.begin(), v.end());
• 내림차순 - sort(v.rbegin(), v.rend());
• 기타 - sort(v.begin(), v.end(), cmp); 가 있다.

여기서 v.begin(), v.end()는 vector v의 시작 주소와, 끝 주소를 뜻한다.  다음은 코딩 순서다.

1. 2번째 인덱스 값 비교하기
2. 2번째 인덱스 값이 동일하다면 사전 순으로 정렬하기

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