[PS, Algorithm] - 모음 사전 (코딩테스트 연습, LEVEL 2)

📒문제

🎈풀이

from itertools import product as prod

def solution(word):
    dictionary = []
    
    for i in range(1, 6):
        dictionary.extend(prod(['A', 'E', 'I', 'O', 'U'], repeat=i))
    dictionary.sort()
    
    for i in range(len(dictionary)):
        if "".join(dictionary[i]) == word:
            return i+1

딱히 어려울 것은 없었고 중복순열을 이용해 문제를 풀었다.
저번에 공부했던 extend함수가 기억이 안나서... 벨로그에 올려놨던 포스트를 잠깐 참조했다.

extend 함수는 리스트를 요소로 넣을 때 리스트 자체를 요소로 넣는 것이 아닌 리스트의 요소를 하나하나씩 꺼내 리스트에 더하는 함수다. 유용하므로 잘 기억해두자!