프로그래머스)모음사전-pyhton

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

프로그래머스

 

from itertools import product


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

 

처음엔 permutions로 모든 조합을 짠 뒤 인덱스를 출력하려고 했다.

하지만 ['A','A','A','A','A','E','E','E'......'U']

로 조합을 짰더니 첫번째인 'A'나 두번째인 'AA'는 생각하지 못했다.

 

결국 검색을 이용했고 , 데카르트곱을 출력하는 product를 이용했다.