프로그래머스 - 문자열 내 마음대로 정렬하기 [파이썬]

greentea31 2022. 6. 17. 09:19

2022. 6. 17. 09:19

코딩테스트 연습 - 문자열 내 마음대로 정렬하기

문자열로 구성된 리스트 strings와, 정수 n이 주어졌을 때, 각 문자열의 인덱스 n번째 글자를 기준으로 오름차순 정렬하려 합니다. 예를 들어 strings가 ["sun", "bed", "car"]이고 n이 1이면 각 단어의 인덱

programmers.co.kr

풀이 과정

파이썬의 lambda식을 이용해서 정렬하여 문제를 해결하였다.

list.sort(key=lambda x:x[n])식은 list의 n번째 인덱스로 리스트를 정렬해준다.

인덱스 n의 문자열이 같은 원소가 여럿일 경우, 사전순으로 배치해야 하므로 먼저 list.sort()로 사전순으로 정렬하고 인덱스 정렬을 시행하였다.

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