[프로그래머스] 타겟 넘버

문제출처

1. 문제

문제 설명

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를 이용해서 풀었다. 모든 숫자가 +인 경우를 탐색한 뒤 숫자 하나씩 -으로 바꿔서 탐색해서 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;
}

위의 코드가 잘 이해되지 않는다면 아래의 블로그를 참조하면 좋을듯 하다.

3. 참고

https://jjnooys.medium.com/%ED%94%84%EB%A1%9C%EA%B7%B8%EB%9E%98%EB%A8%B8%EC%8A%A4-%ED%83%80%EA%B2%9F-%EB%84%98%EB%B2%84-javascript-1d7983d423b5