언어

'문자열 내 마음대로 정렬하기'

땅호720 2023. 12. 13. 18:40

처음에 주석으로 풀었는데, n번째 인덱스 이전의 문자들이 정렬에 고려되지 않는 문제가 생겼다

[반례] ["aea", "ba", "ce", "aee"], 1  -> ["ba", "aea", "aee", "ce"]

 

그렇기 때문에 문자 전체를 뒤에 잇는 것으로 해결하였다.

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