[Python] 문자열 내 마음대로 정렬하기 (프로그래머스 Lv1 파이썬)

문자열로 구성된 리스트 strings와, 정수 n이 주어졌을 때, 각 문자열의 인덱스 n번째 글자를 기준으로 오름차순 정렬하려 합니다. 예를 들어 strings가 ["sun", "bed", "car"]이고 n이 1이면 각 단어의 인덱스 1의 문자 "u", "e", "a"로 strings를 정렬합니다.제한 조건

• strings는 길이 1 이상, 50이하인 배열입니다.
• strings의 원소는 소문자 알파벳으로 이루어져 있습니다.
• strings의 원소는 길이 1 이상, 100이하인 문자열입니다.
• 모든 strings의 원소의 길이는 n보다 큽니다.
• 인덱스 1의 문자가 같은 문자열이 여럿 일 경우, 사전순으로 앞선 문자열이 앞쪽에 위치합니다.

입출력 예

stringsnreturn

["sun", "bed", "car"]	1	["car", "bed", "sun"]
["abce", "abcd", "cdx"]	2	["abcd", "abce", "cdx"]

 

문제 이해

- 간단한 정렬 문제입니다.

- 주어지는 인덱스 값에 따라 우선 정렬하고 이외엔 사전순으로 정렬합니다.

 

문제 풀이

- 람다함수를 사용해서 간결하게 해결할 수 있습니다.

- sort함수는 key값으로 람다함수를 받을 수 있는데, strings의 각 요소를 n에 따라 우선 정렬 후, 나머지를 v에 따라(그냥 사전순)으로 정렬합니다.

- 람다함수에 대한 자세한 내용은 링크를 참고하시면 쉽게 이해할 수 있습니다. def로 정의하지 않고 바로 사용하는 익명함수인데 가끔 한번씩 사용할 경우가 있어서 알아두면 좋습니다.

 

코드

 

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