[Python] 프로그래머스 완전탐색 / 모음 사전

문제 설명
사전에 알파벳 모음 '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번째 단어입니다.

문제해결포인트
Lv2짜리 문제.
1. A,E,I,O,U 로 모든 경우의 수를 구한 리스트를 만든 후 정렬하고
2. 거기서 word의 인덱스를 찾는다.

덕분에 itertools에 product라는 갓갓 함수도 알아간다. 완전 탐색 문제에 대비해서 itertools의 product와 permutations 함수를 따로 정리해놔야지.

여하튼 product 함수를 이용해서 1을 만들고 2를 수행.

코드

from itertools import product

def solution(word):
    dictionary = []
    for i in range(5) : 
        for c in product("AEIOU", repeat=i+1) : 
            dictionary.append("".join(c))
    dictionary.sort()
    return dictionary.index(word)+1