프로그래머스 - 문자열 내 마음대로 정렬하기 - C++

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

프로그래머스

 

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

int wn;

bool compare(string a, string b)
{
    if(a[wn]==b[wn])return a<b;
    else if(a[wn]<b[wn])return true;
    else return false;
}

vector<string> solution(vector<string> strings, int n) {
    vector<string> answer;
    
    wn=n;
    
    sort(strings.begin(),strings.begin()+strings.size(),compare);
    
    return strings;
}

 

주어진 벡터의 문자열들을 n의 인덱스를 기준으로 정렬하는 문제이다.

 

sort 함수에 사용자정의함수를 비교함수로 넣어서 해결하였다. 

n값을 compare함수에 전하기 위해서 전역변수를 만들었다.