문자열로 구성된 리스트 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):
strings.sort()
return sorted(strings, key = lambda x : x[n])
해당 문제에서 밑줄 그어야 할 부분은
각 문자열의 인덱스 n번째 글자를 기준으로 오름차순 정렬하려 합니다.
이 부분에서 lambda
함수를 활용하여 정렬해야 겠다는 아이디어가 떠올라야 합니다.
그리고 두번째 조건으로
인덱스 1의 문자가 같은 문자열이 여럿 일 경우, 사전순으로 앞선 문자열이 앞쪽에 위치합니다.
key의 기준이 동일하다면 사전순으로 표현된것을 앞쪽에 위치하다 ⇒ 배열을 먼저 sort
후 key값을 기준으로 sorting 해주는 아이디어로 해결했습니다.
더 좋은 코드
def solution(strings, n):
return sorted(strings, key = lambda x : (x[n],x))
해당 코드는 (x[n],x)
튜플 형식으로 key값을 부여하여 우선순위를 할당했습니다.