[코딩 테스트 연습(파이썬/Python)] 프로그래머스 _ 문자열 내 마음대로 정렬하기

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

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

[문제 설명]

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

 

[제한 조건]

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

순서대로 하나씩 비교하며 밀어주고 싶었는데, 반복하게 하는데 한참을 고민했다.

재귀함수로 구성해보고 싶었는데 아쉽다.

def solution(strings, n):
    for x in range(len(strings)-1):    
        i = 0
        while i < len(strings)-1:
            if strings[i][n] > strings[i+1][n]:
                strings[i:i+2] = [strings[i+1], strings[i]]
            elif (strings[i][n] == strings[i+1][n]) and (strings[i] > strings[i+1]):
                strings[i:i+2] = [strings[i+1], strings[i]]
            i += 1
    return strings

그보다 sorted의 인자인 key를 활용하고 싶었는데, key자리에 함수가 갈 수 있다는걸 잊어서 활용하지 못했다.

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

위와 같이 key를 함수로 주면, 비교하고자 하는 것들에 해당 함수를 적용하여 정렬할 수 있다.

사전이나, 튜플 형식의 데이터 역시 비슷한 방법으로 정렬이 가능하다.

# 사전/튜플 정렬 예시 _ 알파벳 순으로 정렬
list(enumerate("cdaxbic"))

sorted(list(enumerate("cdaxbic")), key = lambda x: x[1])