문자열로 구성된 리스트 strings와, 정수 n이 주어졌을 때, 각 문자열의 인덱스 n번째 글자를 기준으로 오름차순 정렬하려 합니다. 예를 들어 strings가 ["sun", "bed", "car"]이고 n이 1이면 각 단어의 인덱스 1의 문자 "u", "e", "a"로 strings를 정렬합니다.
| strings | n | return |
|---|---|---|
| ["sun", "bed", "car"] | 1 | ["car", "bed", "sun"] |
| ["abce", "abcd", "cdx"] | 2 | ["abcd", "abce", "cdx"] |
def solution(strings, n):
return sorted(strings, key=lambda x: (x[n], x))
이전에 코테를 공부하며 익혔던 정렬 시 lambda를 key로 활용하는 방법을 적용했다.
조건에서 n번째 단어가 같을 경우 단어 자체를 사전순으로 정렬해야했기 때문에 n번째 수만 보는 것이 아니라 n번째 수가 같으면 단어 자체로 정렬하도록 코드를 작성했다.
벨로그에 업로드하지는 않았지만, 워낙 문제를 풀면서 자주 등장하고 활용했던 개념이라 빠르게 문제를 해결할 수 있어 뿌듯했다!