[프로그래머스 | Python] 모음사전

프로그래머스_모음사전

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

문자열 안에 들어갈 수 있는 문자도 5개, 가능한 문자열의 길이도 5개이므로 완전탐색이 가능하다
중복순열 함수인 itertools.product를 활용하여 가능한 모든 경우의 수를 list로 만든다
A, AA, AAA..순으로 이루어지는 사전순으로 정렬하고, 찾고자하는 word의 index값+1을 return하면 된다