Programers : 문자열 내 마음대로 정렬하기

김정욱·2021년 1월 19일
0

Algorithm - 문제

목록 보기
39/249

sort()에 조건달아 정렬하기

  • compare()에 변수가 필요할 땐 전역으로 선언 후 사용하게 하면 된다!

코드

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

using namespace std;
int index;

bool compare(string a, string b) {
    if (a[index] < b[index]) return true;
    if (a[index] > b[index]) return false;
    return a < b;
}

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

    sort(strings.begin(), strings.end(), compare);
    answer = strings;
    return answer;
}
profile
Developer & PhotoGrapher

0개의 댓글