[PGS] 타겟넘버 - deque 를 이용한 풀이 / Level 2 / 30분 / 성공

Author: Mingguriguri

minjeong/DFSBFS/2024-06-19-[PGS]-#타겟넘버-BFS.py

@@ -0,0 +1,37 @@
+'''
+타겟넘버 - deque를 이용한 BFS 풀이
+'''
+from collections import deque
+
+def solution(numbers, target):
+    answer = 0
+    queue = deque()
+    n = len(numbers)
+    # 초기 상태를 큐에 추가. 시작 숫자에서 +와 - 두 가지 경우를 모두 추가 
+    # (이때 인덱스값도 함께 추가) [값, 인덱스] (enqueue 작업)
+    queue.append([numbers[0], 0])       # 첫 번째 숫자를 더한 경우
+    queue.append([-1*numbers[0], 0])    # 첫 번째 숫자를 뺀 경우
+
+    # 큐가 빌 때까지 반복
+    while queue: 
+        temp, index = queue.popleft() # dequeue. 큐에서 값을 꺼내서 현재 합과 인덱스 가져옴
+        index += 1 # 다음 인덱스로 넘어감
+        # 인덱스가 리스트의 길이보다 작다면,
+        if index < n: 
+            # 현재 합에 다음 숫자를 더하거나 빼는 두 가지 경우를 큐에 추가
+            queue.append([temp + numbers[index], index])
+            queue.append([temp - numbers[index], index])
+        else: # 다 순회했다면 temp 값과 target값 비교
+            if temp == target:
+                answer += 1
+    return answer
+
+# 예제 1: 숫자 배열 [1, 1, 1, 1, 1]로 목표 값 3을 만드는 방법의 수
+numbers = [1, 1, 1, 1, 1]
+target = 3
+print(solution(numbers, target))  # 결과: 5
+
+# 예제 2: 숫자 배열 [4, 1, 2, 1]로 목표 값 4를 만드는 방법의 수
+numbers = [4, 1, 2, 1]
+target = 4
+print(solution(numbers, target))  # 결과: 2