프로그래머스 Lv2 | 완전탐색 | 모음사전

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

프로그래머스

 

그래프로 풀어야하나 싶었는데 중복순열을 이용하면 해결가능했음

from itertools import product

def solution(word):
    vowels = ['A', 'E', 'I', 'O', 'U']
    dict = []
    
    for i in range(1,6):
        for p in product(vowels, repeat=i):
            dict.append(''.join(p))
            
    dict.sort()
    return dict.index(word) + 1

 

dfs로도 풀 수 있음

from itertools import product

def solution(word):
    vowels = ['A', 'E', 'I', 'O', 'U']
    dict = []
    
    def dfs(curr):
        if 1 <= len(curr) <= 5:
            dict.append(curr)
        if len(curr) == 5:
            return
        for vowel in vowels:
            dfs(curr + vowel)
    
    dfs("")
            
    dict.sort()
    return dict.index(word) + 1