Algorithm/프로그래머스

(파이썬) [알고리즘 문제 풀이] - 문자열 내 마음대로 정렬하기 (프로그래머스)

Jesse 2021. 1. 21. 01:06

프로그래머스의 문제 중 하나인 문자열 내 마음대로 정렬하기 입니다.

문자열로 구성된 리스트 strings와 n에 대하여

strings를 n번째 인덱스로 정렬해서 리턴하면 되는 문제입니다.

문제에서 요구하는 정렬을 하기 위해서는 딱 2가지 규칙이 필요합니다.

  • n번째 인덱스를 기준으로 오름차순
  • 동일한 n번째 인덱스가 여럿일 경우 사전 순으로 정렬

사전 순으로 먼저 정렬을 한 후에 n번째 인덱스를 기준으로 정렬을 나중에

하여 문제에서 요구하는 조건을 달성했습니다.

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

파이썬에서는 lambda를 이용하여

특정 인덱스를 기준으로 하는 정렬을 이용할수 있습니다.

문제를 풀다가 필요한 함수를 찾게 되고

또 새로운 것들을 배워 가는것 같네요.