[프로그래머스] 모음 사전

문제

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

프로그래머스

 

풀이

백트래킹을 사용해 풀 수 있었다.

이 문제 같은 경우에는 인자의 개수가 5개 밖에 되지 않기 때문에 파이썬 내장 함수인 product 를 사용하면 더욱 쉽게 풀린다.

 

코드

def solution(word):
    count = 0
    flag = False
    stack = []
    
    def DFS():
        nonlocal count, flag
        if len(stack) < 5:
            for i in ['A', 'E', 'I', 'O', 'U']:
                if not flag:
                    count += 1
                    stack.append(i)
                    if ''.join(stack) == word:
                        flag = True
                        return
                    DFS()
                    stack.pop()
                    
    DFS()
    return count