프로그래머스 level2 모음사전

Problem

Solution

from itertools import product

def solution(word):
    word_list = ['A', 'E', 'I', 'O', 'U']
    
    dictionary = []
    for i in range(1, 6):
        for dict_word in product(word_list, repeat=i):
            dictionary.append("".join(dict_word))
        
    return sorted(dictionary).index(word) + 1

데카르트의 곱을 구할 수 있는 itertools.product()를 이용하여 구하였다.

Reference

https://docs.python.org/ko/3/library/itertools.html#itertools.product