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

문제

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

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

문제풀이

algorithm의 sort를 이용하여 정렬을 하면 된다.

따로 compare함수를 만들어서 idx번째 순으로 정렬을 하게 한다, 둘이 같을 경우 사전 순으로 정렬을 한다.

 

코드

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

bool compare(string s1, string s2){
    if(s1[idx]<s2[idx])
        return true;
    else if(s1[idx]==s2[idx])
        return s1<s2;
    else return false;
}

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