[프로그래머스][level2] 모음사전 - python

문제 링크

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

코딩테스트 연습 - 모음사전

풀이

- 중복 순열 사용

 

1. 1-5개의 모음으로 만들 수 있는  중복 순열 조합 구하고 하나의 문자열로 연결하여 dic에 저장  

2. dic 정렬 

3. 찾고자 하는 단어의 위치 + 1(index는 0부터 시작하기 때문에) 리턴

from itertools import product

def solution(word):
    answer = 0
    
    words= [ 'A', 'E', 'I', 'O', 'U' ]
    dic = [] 
    for i in range(1,6):
        pro = list(product(words,repeat=i))
        
        for j in pro:
            dic.append("".join(j))
    dic.sort()
    
    return dic.index(word) +1