[LV1] 프로그래머스 문자열 내 마음대로 정렬하기 python

프로그래머스로 코딩테스트 뽀개기

정렬 문제인데, key를 어떻게 사용하는지를 보여주는 아주 대표적인 문제이다.

이랑 비슷한 문제가 itertools의 cmp_to_key를 사용하는게 하나 더 있다.

(x[n],x)로 하면 앞에 것을 먼저 적용한 뒤, 같을 경우 뒤 기준을 적용한다.

간단하게 이해하면, 퀵소트 같은 것을 하는데 비교를 대소 대신 이 함수를 쓸 수 있도록 해주는 것이다.

 

파이썬의 기본은 오름차순이다. 기억합시당.

뒤집으려면 reverse=True 인수를 넣어준다.

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