프로그래머스-파이썬 (문자열 내 마음대로 정렬하기)

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

 

 

<제한사항>

 

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

 

<입출력 예시>

 

 

<예시 설명>

 

#입출력 예 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 = []
    n_keyword = []  #문자열의 n번째 단어를 저장해놓을 리스트
    
    for i in strings:
        keyword = i[n]
        n_keyword.append(keyword)
    
    n_keyword.sort()
    
    for i in n_keyword:
        for j in strings:
            if i == j[n] and j not in answer:
                answer.append(j)
    
    return answer

 

각 문자열의 n번 째 단어를 기준으로 주어진 strings를 나열해야한다. 

가장 먼저 떠오른 것은 각 문자열 n번 째에 위치한 단어를 별도로 저장해놓을 리스트가 필요하다 생각했다.

모든 strings의 n번 째 단어를 n_keyword에 저장해놓은 후, 여기에 담긴 단어의 순서 그대로만 string의 n번 째 단어와 

비교하며 동일하면 answer에 추가하면 되는 것 같았다. 

'

'

'

결과는 실패..

n번 째의 단어가 같은 문자열이 여러 개인 경우, n번째 이하의 단어 사전순이 아니라

전체 문자열을 기준으로 사전순 나열해야 했다.

잠시 고민하다가 생각난 것은, 입력받은 문자열 strings 전체를 우선 사전순으로 배열하는 것이었다.

 

def solution(strings, n):
    answer = []
    n_keyword = [] #문자열의 n번째 단어를 저장해놓을 리스트
    strings.sort() #추가!
    
    for i in strings:
        keyword = i[n]
        n_keyword.append(keyword)
    
    n_keyword.sort()
    
    for i in n_keyword:
        for j in strings:
            if i == j[n] and j not in answer:
                answer.append(j)
    
    return answer

 

결과는 성공적이었다.

n번 째의 단어가 동일한 것이 여러 개일경우, 전체 문자를 기준으로 사전순 나열하는 것이기 때문에 strings를 애초에

오름차순 나열할 한 후에 시작하면 그 안에서 순서대로 n번 째의 기준에 맞는 문자열부터 answer에 append되기 때문이다.