[프로그래머스] 타겟 넘버(DFS/BFS) -Java

▤ 목차

1. 문제설명

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 이하인 자연수입니다.

입출력 예

numbers	target	return
[1, 1, 1, 1, 1]	3	5
[4, 1, 2, 1]	4	2

입출력 예 설명
입출력 예 #1
문제 예시와 같습니다.

입출력 예 #2
+4+1-2+1 = 4
+4-1+2-1 = 4
총 2가지 방법이 있으므로, 2를 return 합니다.

2. 문제풀이

나의 풀이 (DFS 풀이)

class Solution {
    private int answer = 0;
    public int solution(int[] numbers, int target) {
        
        dfs(numbers, target, 0 ,0);
        
        return answer;
    }
    
    public void dfs(int[] numbers, int target, int index, int currentSum){
        // 모든 숫자를 다 사용했을때
        if(index == numbers.length){
            //타겟과 일치할때
            if(currentSum == target){
                answer++;
            }
            return;
       }
       // 다음 숫자를 더하는 경우
       dfs(numbers, target, index+1 , currentSum + numbers[index]);
        
       // 다음 숫자를 빼는 경우
       dfs(numbers, target, index+1 , currentSum - numbers[index]);
        
    }
}

다른 풀이 (BFS 풀이)

import java.util.LinkedList;
import java.util.Queue;

class Solution_BFS {
    public int solution(int[] numbers, int target) {
        int answer = 0;
        int len = numbers.length;
        // 큐에는 현재까지의 합계를 저장
        Queue<Integer> queue = new LinkedList<>();

        // 1. 초기 상태: numbers[0]을 더한 값과 뺀 값 두 가지를 큐에 넣음
        queue.offer(numbers[0]);
        queue.offer(-numbers[0]);

        // 2. 두 번째 숫자부터 마지막 숫자까지 반복 (이미 첫 번째 숫자는 처리됨)
        for (int i = 1; i < len; i++) {
            int size = queue.size();
            
            // 현재 큐에 있는 모든 합계에 대해 다음 숫자를 적용
            for (int j = 0; j < size; j++) {
                int currentSum = queue.poll();
                
                // 현재 숫자를 더하는 경우
                queue.offer(currentSum + numbers[i]);
                
                // 현재 숫자를 빼는 경우
                queue.offer(currentSum - numbers[i]);
            }
        }

        // 3. 모든 숫자를 처리한 후, 큐에 남아있는 최종 합계 중 타겟 넘버와 일치하는 것 세기
        while (!queue.isEmpty()) {
            if (queue.poll() == target) {
                answer++;
            }
        }

        return answer;
    }
}

잘못된 내용이 있다면 지적부탁드립니다. 방문해주셔서 감사합니다.