[프로그래머스] Lv.2 모음사전 [Python, 파이썬]

• 문제 : https://school.programmers.co.kr/learn/courses/30/lessons/84512
• 🔑 DFS로 완전 탐색
  • A ~ UUUUU까지 만들 수 있는 dfs 함수를 생성
  • 원하는 단어를 찾을 때까지 answer 올리고
  • 찾았으면 계속 리턴만해서 빠져나오기

 

def dfs(w, word, words):
    global answer, flag
    
    if word == w:
        flag = True
        return
    
    if len(w) != 5:
        for i in range(len(words)):
            if flag:
                return
            answer += 1
            dfs(w+words[i], word, words)
            
answer = 0
flag = False

def solution(word):

    words = ['A', 'E', 'I', 'O', 'U']
                
    dfs("", word, words)
    
    return answer