[Level1] 문자열 내 마음대로 정렬하기 (C++/Python)

C++)

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

bool compare(string& str1, string& str2) {
	// 비교하고자 하는 위치의 문자가 같다면
    if(str1.at(index) == str2.at(index))
        return str1 < str2;	// 오름차순 정렬
    // 아니라면
    else
        return str1.at(index) < str2.at(index);	// 비교하는 문자 기준 오름차순 정렬
}

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

    sort(strings.begin(), strings.end(), compare);
    
    return strings;
}

 

 

 

 

Python)

def solution(strings, n):
    strings.sort()
    strings.sort(key=lambda st : st[n])
    return strings