알고리즘 풀이/프로그래머스

[Level 1] 문자열 내 마음대로 정렬하기

iwannawebfullstack 2021. 2. 19. 00:02

programmers.co.kr/learn/courses/30/lessons/12915

 

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

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

programmers.co.kr


접근

파이썬의 sort()에 key값을 넣어서 정렬을 하였다.

만약에 n번째 숫자가 같다면 사전순으로 정렬해야하기 때문에 먼저 기본적인 sort를 진행한 후에 key값을 추가하여 n번째 글자로 정렬되도록 하였다.

 

구현

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

 

'알고리즘 풀이 > 프로그래머스' 카테고리의 다른 글

[Level 1] 비밀 지도  (0) 2021.02.19
[Level 1] 2016년  (0) 2021.02.19
[Level 1] 크레인 인형 뽑기  (0) 2021.02.18
[고득점 KIT] 프린터 cpp  (0) 2021.02.12
[고득점 KIT] 다리를 지나는 트럭 cpp  (0) 2021.02.11