[프로그래머스] 타겟넘버 (Lv 2) - javascript

문제

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

풀이

DFS(깊이 우선 탐색) 알고리즘으로 풀어야 하는 문제다.
 

function solution(numbers, target) {

    let answer = 0; // target과 일치하는 횟수

    dfs(0,0); 

    function dfs(level, sum) {
        if(level === numbers.length) { // 1. 재귀 종료 조건 설정
            if(sum === target) answer += 1; // target과 일치하면 answer +1
            return; // 재귀 함수 종료
        }
        dfs(level + 1, sum + numbers[level]); // 2. 재귀함수 호출 (+) 
        dfs(level + 1, sum - numbers[level]); // 3. 재귀함수 호출 (-)
    }

    return answer; // 결과 반환
}

 
numbers = [2,2,2,2]로 예시를 들고 코드 진행 과정을 살펴보자.
 

1. 맨 처음 level 0, sum 0에서 시작
2. 1번 종료 조건을 통과해 재귀함수가 호출(2) -> level 1, sum 2 로 함수 진행
3. level과 numbers.length가 같지 않으므로 끝까지 재귀함수 호출 -> ...
4. level 4, sum 8 이 되면 재귀 종료 조건과 일치
5. sum 과 target 비교해서 명령문 실행 -> 재귀함수 종료
   여기서 return을 안 써주면 계속 재귀함수를 실행하기 때문에 에러 발생
6. level 3, sum 6 시점으로 돌아가서 아래 3번 재귀함수 실행
7. level 4, sum 4 가 되어 종료 조건과 만남.... 반복
    

이런 식으로 깊숙히 내려왔다가 다시 올라갔다 내려왔다 하면서 종료 조건과 만남을 반복하게 된다.
그 과정에서 sum = target이면 answer에 숫자가 쌓이게 된다.
 

자세한 설명
 

• 정확도 테스트 결과
  

  

  
   

function solution(numbers, target) {

    let answer = 0;

    dfs(0,0); 

    function dfs(level, sum) {

        if(level < numbers.length) {
            dfs(level + 1, sum + numbers[level]); 
            dfs(level + 1, sum - numbers[level]); 
        }
        else {
            if(sum === target) answer += 1;
        }
    }
    return answer;
}

 
이렇게 작성하면 이해가 더 쉽지만, 속도가 살짝 느려진다.
 

---

참고

https://jjnooys.medium.com/