코딩테스트 연습 > 깊이/너비 우선 탐색(DFS/BFS) > 타겟 넘버
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 함수를 작성해주세요.
numbers | target | result |
---|---|---|
[1, 1, 1, 1, 1] | 3 | 5 |
[4, 1, 2, 1] | 4 | 2 |
class Solution {
int count = 0;
public int solution(int[] numbers, int target) {
dfs(numbers, 0, target, 0);
return count;
}
public void dfs(int[] numbers, int depth, int target, int temp){
if(depth == numbers.length){
if(target == temp) {
count ++;
}
return;
}
int plus = temp + numbers[depth];
int minus = temp - numbers[depth];
dfs(numbers, depth+1, target, plus);
dfs(numbers, depth+1, target, minus);
}
}
깊이 우선 탐색 (DFS) 알고리즘 이용
아래 그림을 먼저 이해하면 코드 구현이 조금 쉬워진다...!!!
int count
: 전역 변수int depth
: 이진트리 깊이 (numbers
배열의 인덱스)depth == numbers.length
int temp
: 이전 노드까지의 합 dfs(int[] numbers, int depth, int target, int temp)
: 깊이 우선 탐색을 위한 재귀 함수