[프로그래머스/Level1/파이썬3(python3)] 문자열 내 마음대로 정렬하기

[프로그래머스/Level1/파이썬3(python3)] 문자열 내 마음대로 정렬하기

문제

문자열로 구성된 리스트 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):
    answer = []
    for string in strings:
        answer.append(string[n] + string)
    answer.sort()
    answer = [x[1:] for x in answer]
    return answer

설명

다른 풀이들을 보니 보통 sorted(다른 기준으로 정렬하는 함수)함수로 풀었는데 나는 그게 생각이 안났어가지고 내맘대로 풀었다. 정렬해야할 글자를 맨 앞에 가져와서 문자열을 정렬한 뒤에 그 글자를 다시 없애는 식으로 해결하였다. 첫번째 예제로 설명하면 sun, bed, car가 usun, ebed, acar가 되고 이를 정렬하면 acar, ebed, usun이 되는데 여기에서 첫번째 글자만 없애면 car, bed, sun이라는 단어들만 남아 정렬이 완료된다.