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

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr

풀이

compare 함수를 작성하여 해결하였다.

코드

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

int N;

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

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