[프로그래머스] Lv.2 타겟 넘버 (파이썬) _ 깊이/너비 우선탐색(DFS/BFS)

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

프로그래머스

문제 설명

n개의 음이 아닌 정수들이 있습니다. 이 정수들을 순서를 바꾸지 않고 적절히 더하거나 빼서 타겟 넘버를 만들려고 합니다. 예를 들어 [1, 1, 1, 1, 1]로 숫자 3을 만들려면 다음 다섯 방법을 쓸 수 있습니다.

-1+1+1+1+1 = 3
+1-1+1+1+1 = 3
+1+1-1+1+1 = 3
+1+1+1-1+1 = 3
+1+1+1+1-1 = 3

사용할 수 있는 숫자가 담긴 배열 numbers, 타겟 넘버 target이 매개변수로 주어질 때 숫자를 적절히 더하고 빼서 타겟 넘버를 만드는 방법의 수를 return 하도록 solution 함수를 작성해주세요.

제한사항

• 주어지는 숫자의 개수는 2개 이상 20개 이하입니다.
• 각 숫자는 1 이상 50 이하인 자연수입니다.
• 타겟 넘버는 1 이상 1000 이하인 자연수입니다.

---

👣 1. DFS (깊이 우선 탐색)

“한 방에 들어가면 끝까지 들어가고, 더 이상 못 가면 되돌아와서 다른 길로 간다”

 

• 우선 하나의 경로를 깊게 파고든다

• 나중에 갈 수 있는 방이 있어도 일단 지금 갈 수 있는 방부터 깊숙이 들어감

• 스택(stack) 또는 재귀(recursion)로 구현

 

💬 비유: 미로에서 갈 수 있는 길이 있으면 쭉쭉 들어가고, 막히면 되돌아와서 다른 길 찾는 느낌!

 

---

👥 2. BFS (너비 우선 탐색)

“현재 위치에서 갈 수 있는 방을 전부 먼저 둘러본 다음, 그 다음 층으로 넘어간다”

 

• 현재 레벨(거리)에서 갈 수 있는 방을 전부 탐색

• 그 다음 레벨로 넘어감

• 큐(queue)를 사용해서 구현

 

💬 비유: 건물 1층 방 다 보고, 그 다음 2층 방, 그 다음 3층 방… 층별로 차근차근 보는 느낌!

---

1. DFS 풀이

def solution(numbers, target):
    answer = 0
    
    def dfs(index, current_sum):
        nonlocal answer
        
        #종료 조건
        if index == len(numbers):
            if current_sum == target:
                answer += 1
            return  #종료한다는 뜻, 인덱스가 끝까지 갔는데 값이 나올 수 없다면!
        
        dfs(index+1, current_sum + numbers[index])
        dfs(index+1, current_sum - numbers[index])
    
    dfs(0, 0)
    return answer

• 재귀함수? 먼데여 ;;

 

2. BFS 풀이

from collections import deque

def solution(numbers, target):
    queue = deque()
    queue.append((0,0))
    count = 0
    
    while queue:
        index, current_sum = queue.popleft()
        if index == len(numbers):
            if current_sum == target:
                count += 1
        
        else:
            queue.append((index+1, current_sum+numbers[index]))
            queue.append((index+1, current_sum-numbers[index]))
    
    return count

• 큐(queue) 자료구조를 배웠다. FIFO(First In First Out) 방식.

 

---

왜 이렇게 폭풍 이론 공부를 하냐,, 내일모레 코테를 보거등요,, 망했다,,,