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

문제

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

 

코드

#include <string>
#include <vector>
#include <algorithm>

using namespace std;


int index;

bool cmp(string a,string b){
    if(a[index]!=b[index]) return a[index]<b[index];
    else return a<b;
}



vector<string> solution(vector<string> strings, int n) {
    vector<string> answer;

    index=n;

    sort(strings.begin(),strings.end(),cmp);

    answer=strings;

    return answer;
}