[프로그래머스] Lv 2 모음사전 - Python

1. 문제

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

프로그래머스

사전에 알파벳 모음 'A', 'E', 'I', 'O', 'U'만을 사용하여 만들 수 있는, 길이 5 이하의 모든 단어가 수록되어 있습니다. 사전에서 첫 번째 단어는 "A"이고, 그다음은 "AA"이며, 마지막 단어는 "UUUUU"입니다.

단어 하나 word가 매개변수로 주어질 때, 이 단어가 사전에서 몇 번째 단어인지 return 하도록 solution 함수를 완성해주세요.

 

2. 제한사항 

 

3. Python code

from itertools import product
 
def solution(word):
    dictionary = []
    for i in range(1,6):
        for j in product('AEIOU', repeat=i):
            dictionary.append(''.join(list(j)))
    dictionary.sort()
 
    return dictionary.index(word) +1

중복 순열이라 itertools에 있는 product를 사용해 풀면 쉽게 풀린다. 

 

product(리스트, repeat=n) 형식으로 사용하면 된다. 

product는 데카르트의 곱으로 정렬된 튜플 형식의 값을 반환한다.

 

리스트 [A, E, I, O, U]를 repeat 1부터 5까지 돌려준다.

repeat=1일때는 글자 1개(A, E, I, O, U), repeat=2일 때는 글자 두개(AA, AE, ...), repeat=5(AAAAA, ..., UUUUU)까지

처음에 선언해준 리스트에 append시킨다. 

주어진 word가 몇번째 인덱스에 있는지 찾기 위해 리스트를 정렬해준다.