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

문제 링크

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

문제 설명

문자열을 담은 리스트 strings와 정렬의 기준이 될 인덱스 n이 주어지면, 각 문자열의 n번 문자를 기준으로 정렬된 리스트를 반환해라.

1차 시도 - 틀림

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

처음엔 어떻게 할지 감이 안왔었는데 생각해보니 sort에 기준을 설정해줄 수 있었던 것 같아서 찾아보니 sort, sorted에 key인자를 전달해줌으로써 기준을 정할 수 있었다.

그래서 맞는줄 알았는데 불통.
왜인지 return값을 살펴보니 문제에서의
인덱스 1의 문자가 같은 문자열이 여럿 일 경우, 사전순으로 앞선 문자열이 앞쪽에 위치합니다.
이 부분을 충족시키지 못했기 때문이었다.
key 기준으로 sort하면 나머지 요소들은 영향을 끼치지 못하고 기존 순서를 유지하기 때문이다.

2차 시도

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

그래서 return 전에 sort를 한 번 해주면 통과!

다른 사람의 풀이를 보니 sort를 쓰지 않고 key를 튜플로 해줌으로써 정렬의 우선순위를 지정해주는 방법도 있다.

다른 사람의 풀이 참고

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

이런식으로!
lambda에 튜플을 넣어줌으로써 sort의 우선순위를 정할 수 있다.
어메이징 파이썬

정리

1. sort, sorted에 key인자를 넣어줌으로써 정렬 기준을 정할 수 있다.
2. key 기준으로 정렬을 할 경우 오직 key 기준으로만 정렬을 한다. key이외의 나머지 요소에 대해선 정렬되지 않음.
3. lambda를 이용한 key 설정에서 튜플을 넣어줌으로써 정렬의 우선순위를 지정할 수 있다.