[프로그래머스] 모음사전 - 파이썬/구현

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

from itertools import product

def solution(word):
    answer = 0
    word_list = ['A','E','I','O','U','']
    pro_list = []

    A = list(product(word_list,repeat=5))

    for i in A:
        C = ''.join(i)
        pro_list.append(C)

    pro_list = list(set(pro_list))
    pro_list.sort()

    answer = pro_list.index(word)

    return answer

모음과 더불어 공백문자를 리스트에 넣어주고 product를 해준다.

그러면 A는 A와 공백 4칸을 선택하는 것으로 표현할 수 있다.

그렇게 만든 리스트를 정렬하여 인덱싱해준다.