[프로그래머스 Lv.1] 문자열 내 마음대로 정렬하기

문제는 다음과 같습니다.

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

프로그래머스

 

풀이 코드는 다음과 같습니다.

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

 

i번째 단어로 정렬하는 방법은 sort함수에 비교함수를 직접 만들어 사용하는 방법입니다. 비교하는 cmp함수를 만들고 sort함수 안에 넣어 i번째 단어를 기준으로 정렬할 수 있도록 할 수 있습니다.