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 | return |
---|---|---|
[1, 1, 1, 1, 1] | 3 | 5 |
[4, 1, 2, 1] | 4 | 2 |
문제 예시와 같습니다.
+4+1-2+1 = 4
+4-1+2-1 = 4
DFS를 이용해서 풀었다. 모든 숫자가 +
인 경우를 탐색한 뒤 숫자 하나씩 -
으로 바꿔서 탐색해서 target
의 값이 나오면 cnt
를 증가시켰다.
처음에는 dfs
를 실행하기 전에 모든 수를 더했을 때 target
이 되는 경우를 미리 처리하려고 했는데 오히려 시간이 더 오래걸렸다. 해당 부분은 아래와 같고 dfs로 탐색하면 맨 처음으로 처리되는 부분이라서 없어도 되는 코드다.
if (numbers.reduce((acc, cur) => acc+=cur, 0) === target) return 1
전체 코드는 아래와 같다.
function solution(numbers, target) {
let cnt = 0;
function dfs(idx, acc) {
if (idx === numbers.length) {
if (acc === target) cnt++;
return;
}
dfs(idx + 1, acc + numbers[idx]);
dfs(idx + 1, acc - numbers[idx]);
}
dfs(0, 0);
return cnt;
}
위의 코드가 잘 이해되지 않는다면 아래의 블로그를 참조하면 좋을듯 하다.