[Python] 프로그래머스 12915. 문자열 내 마음대로 정렬하기 (Lv.1)

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

프로그래머스

 

lambda를 이용한 정렬을 할 줄 안다면 정말 쉽게 풀 수 있는 문제입니다.

단어의 n번째 글자를 기준으로 정렬해줍니다. 만약 n번째 글자가 같다면 단어를 기준으로 정렬해 줍니다.

 

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

 

sort(key=lambda x: (x[n], x))

 

정렬의 key를 lambda식으로 표현해 주는데, 우선 순위는 x[n] 즉 어떤 단어 x의 n번째 글자이고 그 값이 같다면 x 즉 어떤 단어 자체를 기준으로 정렬하겠다는 뜻입니다.