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

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

문제 예시와 같습니다.

입출력 예 #2

+4+1-2+1 = 4
+4-1+2-1 = 4

• 총 2가지 방법이 있으므로, 2를 return 합니다.

 

구현코드 & 설명(BFS)

import java.util.*;


class Solution {


    public int solution(int[] numbers, int target) {
        Queue<Status> que = new LinkedList<Status>();
        int[] cal = {1, -1}; // 합, 차를 담는 배열
        int method =0; // 방법의 갯수를 담는 변수
        
         for(int i=0; i< cal.length; i++){
				//첫 노드 = > 1* numbers[0] 
                //첫 노드 = > -1* numbers[0]
                // que에 넣는다
                int newSum =  cal[i]*numbers[0];
                que.add(new Status(newSum, 0));
            }

        
        while(!que.isEmpty()){
            Status st =que.poll();
            
            // 만약 이미 모든 노드를 탐방했다면
            if(st.count==numbers.length -1) {
                // 타겟과 현재까지의 sum이 같은지 확인, 같다면 method ++
                if(st.sum == target) method++;
                continue;
                
            }
            
            // 만약 모든 노드를 탐방하지 않았다면
            // 그 다음 자식 노드들 다시 탐방 (합, 차 노드)
            for(int i=0; i< cal.length; i++){
                int newSum = st.sum + cal[i]*numbers[st.count+1];
                que.add(new Status(newSum, st.count+1));
            }
            
        }  
        
        return method;
    }
}
  
class Status{
    int sum; // 현재까지 합
    int count; // 몇번째까지 더했는지, 인덱스
    
    public Status(int sum, int count){
        this.sum =sum;
        this.count =count;
    }
    
}