[Programmers - lv01] 문자열 내 마음대로 정렬하기 (cpp / python)

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

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

문제 설명

내장되어있는 정렬함수를 이용할 줄 아는지 물어보는 문제

문제 풀이

소스코드 : C++

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

int N;
bool comp(string a, string b){
    if(a[N]==b[N])return a<b;
    else return a[N]<b[N];
}
vector<string> solution(vector<string> strings, int n) {
    N=n;
    sort(strings.begin(),strings.end(),comp);
    return strings;
}

소스코드 : Python

def solution(strings, n):
    return sorted([s for s in strings],key = lambda a:[a[n],a])