모음 사전

코딩테스트 연습 - 모음사전 | 프로그래머스 스쿨 (programmers.co.kr)

프로그래머스

 

 

나의 코드

from itertools import product

def solution(word):

    answer = 0

   

    #사전 리스트

    word_list = []

    for i in range(1,6):

        #중복순열 돌리기

        for j in product(['A','E','I','O','U'], repeat=i):

            word_list.append("".join(j))

    # 사전순으로 정렬

    word_list.sort()

   

    # 순서 뽑아내기 index로 뽑아내기 떄문에 +1 해줘야됨

    answer = word_list.index(word)+1

    #print(answer)

    return answer

solution("AAAAE")