[python/알고리즘] 프로그래머스 문자열 마음대로 정렬하기

·2024년 11월 30일
0

코드카타 47번 - 문자열 마음대로 정렬하기

한 줄 코드

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

sorted(_, key = __)

  • sorted()에서는 key를 통해 정렬 기준을 정할 수 있다.
  • sorted(list, key = len) : 원소의 길이 순으로 정렬
  • sorted(list, key = abs) : 원소의 절댓값 크기로 정렬 등

이 문제에서는 key에 lambda를 이용해 인덱스 n의 값을 넣어줄 수 있었다.
튜플 (x[n], x)로 묶은 것은 (1) 우선은 인덱스 n의 값을 기준으로 정렬하고 그 값이 같은 경우 (2) 전체 단어 순으로 정렬해야 하기 때문이다.

lambda

  • lambda 인자: 반환값

lambda, map, filter 복습하기 from hye.velog
파이썬 공식 문서 - Sorting Techniques

항상 공식 문서부터 살펴보는 습관을 기르자
구글링 잘 해보기

profile
To Dare is To Do

0개의 댓글