[프로그래머스/파이썬] 모음사전

문제 설명

사전에 알파벳 모음 'A', 'E', 'I', 'O', 'U'만을 사용하여 만들 수 있는, 길이 5 이하의 모든 단어가 수록되어 있습니다. 사전에서 첫 번째 단어는 "A"이고, 그다음은 "AA"이며, 마지막 단어는 "UUUUU"입니다.

단어 하나 word가 매개변수로 주어질 때, 이 단어가 사전에서 몇 번째 단어인지 return 하도록 solution 함수를 완성해주세요.

제한사항

• word의 길이는 1 이상 5 이하입니다.
• word는 알파벳 대문자 'A', 'E', 'I', 'O', 'U'로만 이루어져 있습니다.

---

 

입출력 예

word	result
"AAAAE"	6
"AAAE"	10
"I"	1563
"EIO"	1189

 

입출력 예 설명

입출력 예 #1

사전에서 첫 번째 단어는 "A"이고, 그다음은 "AA", "AAA", "AAAA", "AAAAA", "AAAAE", ... 와 같습니다. "AAAAE"는 사전에서 6번째 단어입니다.

입출력 예 #2

"AAAE"는 "A", "AA", "AAA", "AAAA", "AAAAA", "AAAAE", "AAAAI", "AAAAO", "AAAAU"의 다음인 10번째 단어입니다.

입출력 예 #3

"I"는 1563번째 단어입니다.

입출력 예 #4

"EIO"는 1189번째 단어입니다.

 

 

 

 

 

from itertools import product

def solution(word):
    compare = ['A', 'E', 'I', 'O', 'U']
    word_list = []
    
    for i in range(1, 6):
        word_list.extend(map("".join, product(compare, repeat=i))) 
    
    word_list.sort()
    
    for num, res in enumerate(word_list):
        if (res == word):
            return num+1
        

사실 5글자 밖에 안돼서 그냥 리스트를 만들고 정렬을 해 일치하는 인덱스 번호를 찾았습니다. (좀 무식한 방법이죠… 하하)

 

 

 

문제에서 의도한 풀이는 아래와 같은 풀이가 아닐까 싶습니다.

 

1의 자리: 한 글자인 경우 → 1개

2의 자리: 두 번째 글자가 올 때마다 추가적으로 만들 수 있는 단어 개수 → 5개 ⇒ 총 6개

3의 자리: 세 번째 글자가 올 때마다 추가적으로 만들 수 있는 단어 개수 → 25개 ⇒ 총 31개

4의 자리: 네 번째 글자가 올 때마다 추가적으로 만들 수 있는 단어 개수 → 125개 ⇒ 총 156개

5의 자리: 다섯 번째 글자가 올 때마다 추가적으로 만들 수 있는 단어 개수 → 625개 ⇒ 총 781개

 

 

 

예를 들어 EIO가 사전에 몇 번째로 등장하는지 구한다고 하면

1. 첫 번째 문자 'E'는 'A' 다음 두 번째 문자이므로
   • 'A'로 시작하는 단어는 781개이므로
   • 'E'로 시작하는 단어는 781개 이후부터 시작
2. 두 번째 문자 'I'는 'E' 다음 세 번째 문자이므로
   • 'EA', 'EE', 'EI', 'EO', 'EU' 이런 식으로 5개 그룹이 생김
   • 'EA' 그룹은 156개
   • 'EE' 그룹은 156개
   • 'EI'는 세 번째 그룹이므로 2 * 156 = 312개 이후부터 시작
3. 세 번째 문자 'O'는 'I' 다음 네 번째 문자이므로
   • 'EIA', 'EIE', 'EII', 'EIO', 'EIU' → 각각 31개씩 가짐
   • 'EIA' → 31개
   • 'EIE' → 31개
   • 'EII' → 31개
   • 'EIO' → 3번째 그룹, 즉 3 * 31 = 93개 이후부터 시작

(1 * 781) + (2 * 156) + (3 * 31) + 1 = 1189번째

 

위 규칙(1, 5, 25, 125, 625)은 각 자리에서 만들어질 수 있는 단어 개수가 됩니다. 따라서 각 자리의 위치에 따라 곱하여 순서를 구하는 방식이 필요합니다.

 

def solution(word):
    compare = ['A', 'E', 'I', 'O', 'U']
    weights = [781, 156, 31, 6, 1]
    
    answer = 0
    for i, char in enumerate(word):
        index = compare.index(char)
        answer += index * weights[i] + 1 

    return answer

이런식으로 코드를 만들 수도 있네요…