문자열 내 마음대로 정렬하기 (with Python3)

Phlano 2020. 7. 17. 17:09

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

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

programmers.co.kr

n 인덱스에 있는 요소 값으로 정렬하는 문제이다.

만약 n번째 인덱스가 같다면 전체 문자열을 사전 순으로 순서를 매긴다.

그래서 먼저 전체 리스트를 정렬을 해준다.

(미리 정렬해준다면 n번째 인덱스가 같은 값이 나와서 아무 동작이 안되더라도 사전 순으로 될 수 있다.)

마지막으로 sorted함수에 key에 n번째 인덱스로 정렬하라고 lambda식을 넣어 준다.

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