[파이썬] 문자열 내 마음대로 정렬하기

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의 문자가 같은 문자열이 여럿 일 경우, 사전순으로 앞선 문자열이 앞쪽에 위치합니다.

입출력 예

 

strings	n	return
[sun, bed, car]	1	[car, bed, sun]
[abce, abcd, cdx]	2	[abcd, abce, cdx]

 

입출력 예 설명

입출력 예 1
sun, bed, car의 1번째 인덱스 값은 각각 u, e, a 입니다. 이를 기준으로 strings를 정렬하면 [car, bed, sun] 입니다.

입출력 예 2
abce와 abcd, cdx의 2번째 인덱스 값은 c, c, x입니다. 따라서 정렬 후에는 cdx가 가장 뒤에 위치합니다. abce와 abcd는 사전순으로 정렬하면 abcd가 우선하므로, 답은 [abcd, abce, cdx] 입니다.

 

 

🧸 나의 풀이

 

def solution(strings, n):
    str=[]
    for i in strings:
        str.append(i[n]+i)       # n번째 문자를 앞에 붙임
    str.sort()
    return [i[1:] for i in str]

 

strings의 n번째 인덱스 값을 문자열 맨 앞에 붙이고 정렬한다. 반환할 때 반복문을 이용해서 붙였던 문자를 제거한다.

 

🌞 다른 사람 풀이

 

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

 

lambda함수를 이용해서 x에서 n번째 값을 기준으로 정렬한다. lambda함수를 잘 활용할 수 있도록 해야겠다.

 

def strange_sort(strings, n):
    def sortkey(x):
        return x[n]
    strings.sort(key=sortkey)
    return strings

 

sortkey함수를 만들어서 정렬할 때 key로 쓴다. 

 

 

🌼 해결과정

1. 리스트에서 각 값의 n부터만 저장 후 정렬한다. 음... 잘린 부분은 어떻게 붙이냐??

2. 해쉬를 쓸까

hash=sorted(hash, key=hash.values())

해쉬 key는 완전한 문자열, values는 n번째 인덱스 값으로 해서 values를 비교해서 정렬할려고 했는데 TypeError: 'dict_values' object is not callable

안되는 이유는key specifies a function of one argument that is used to extract a comparison key from each element in iterable (for example, key=str.lower).

다른 사람 블로그를 참조하여 그냥 앞에 갖다 붙이면 되겠구나 아이디어를 얻음