풀이 방법

itertools의 product는 데카르트 곱이다.
모든 경우의 수를 문자열로 만들어주고, set 자료구조의 합집합 연산자를 이용해서 하나의 리스트로 만들어준후, 정렬을 해준다.
문자열의 정렬 성질과 문제의 순서가 같다.

풀이 코드

from itertools import product
def solution(word):
    answer = 0
    
    w = ['A', 'E', 'I', 'O', 'U']
    ls = set()
    for i in range(1, 6):
        ls |= set(map("".join, product(w, repeat = i)))
        
    ls = sorted(list(ls))
    
    return ls.index(word) + 1