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

https://school.programmers.co.kr/learn/courses/30/lessons/84512

from itertools import product

def solution(word):
    lst = []

    for i in range(1, 6):
        for j in product(['A','E','I','O','U'], repeat = i): # repeat의 길이만큼 중복순열 구하기
            w = ''.join(j) # 튜플상태로 반환돼서 join으로 합치기
            lst.append(w)

    lst.sort()
    answer = lst.index(word) + 1 

    return answer

itertools로 순열, 조합만 이용해봤지 중복순열은 처음 사용해봤다.

공식문서 예시

product('ABCD', 'xy') --> Ax Ay Bx By Cx Cy Dx Dy
product(range(2), repeat=3) --> 000 001 010 011 100 101 110 111