Programmers) level2. 모음 사전

5개 밖에 안되는 알파벳이 주어졌으므로 itertools 내장 모듈의 중복 순열 product를 이용해본다.

from itertools import product
def solution(word):
    answer = 0
    lst = []
    for i in range(1, 6):
        for j in product(['A', 'E', 'I', 'O', 'U'], repeat = i):
            lst.append(''.join(j))
    lst.sort()
    return lst.index(word) + 1