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

김서연·2024년 2월 9일

코딩테스트

목록 보기
7/31
post-thumbnail

📜문제 설명

문제 바로가기

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

📍제한 사항

  • strings는 길이 1 이상, 50이하인 배열입니다.
  • strings의 원소는 소문자 알파벳으로 이루어져 있습니다.
  • strings의 원소는 길이 1 이상, 100이하인 문자열입니다.
  • 모든 strings의 원소의 길이는 n보다 큽니다.
  • 인덱스 1의 문자가 같은 문자열이 여럿 일 경우, 사전순으로 앞선 문자열이 앞쪽에 위치합니다.

📍입출력 예시

stringsnreturn
["sun", "bed", "car"]1["car", "bed", "sun"]
["abce", "abcd", "cdx"]2["abcd", "abce", "cdx"]

📄문제 해결

📝내가 푼 코드

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

이전에 코테를 공부하며 익혔던 정렬 시 lambda를 key로 활용하는 방법을 적용했다.

조건에서 n번째 단어가 같을 경우 단어 자체를 사전순으로 정렬해야했기 때문에 n번째 수만 보는 것이 아니라 n번째 수가 같으면 단어 자체로 정렬하도록 코드를 작성했다.


🤔느낀점

벨로그에 업로드하지는 않았지만, 워낙 문제를 풀면서 자주 등장하고 활용했던 개념이라 빠르게 문제를 해결할 수 있어 뿌듯했다!

profile
가보자고! 🔥

0개의 댓글