문제 설명
문자열로 구성된 리스트 strings와, 정수 n이 주어졌을 때, 각 문자열의 인덱스 n번째 글자를 기준으로 오름차순 정렬하려 합니다. 예를 들어 strings가 ["sun", "bed", "car"]이고 n이 1이면 각 단어의 인덱스 1의 문자 "u", "e", "a"로 strings를 정렬합니다.
제한 조건
- strings는 길이 1 이상, 50이하인 배열입니다.
- strings의 원소는 소문자 알파벳으로 이루어져 있습니다.
- strings의 원소는 길이 1 이상, 100이하인 문자열입니다.
- 모든 strings의 원소의 길이는 n보다 큽니다.
- 인덱스 1의 문자가 같은 문자열이 여럿 일 경우, 사전순으로 앞선 문자열이 앞쪽에 위치합니다.
입출력 예
| 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값을 부여하여 우선순위를 할당했습니다.
