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

최지나·2023년 12월 8일
2

코딩테스트

목록 보기
96/154

문제

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

입출력 예

numberstargetreturn
[1, 1, 1, 1, 1]35
[4, 1, 2, 1]42

문제 출처

https://school.programmers.co.kr/learn/courses/30/lessons/43165

생각

  • 숫자들을 더하거나 빼는 모든 경우의 수를 확인해서 그 중 누적 합이 target과 동일한 경우의 수를 count해야 하므로 완전탐색 문제이다 => DFS를 사용하자
  • findWaysToMakeTarget(numbers, target, idx + 1, currentSum + numbers[idx]) 숫자를 더하는 경우 탐색
  • findWaysToMakeTarget(numbers, target, idx + 1, currentSum - numbers[idx]) 숫자를 빼는 경우 탐색
  • 마지막 숫자까지 탐색한 경우(idx == numbers.length) target과 누적합이 같으면 최종 답에 +1

코드

class Solution {
    public int solution(int[] numbers, int target) {
        return findWaysToMakeTarget(numbers, target, 0, 0);
    }

    private int findWaysToMakeTarget(int[] numbers, int target, int idx, int currentSum) {

        if (idx == numbers.length) {
            return currentSum == target ? 1 : 0;
        }

        return findWaysToMakeTarget(numbers, target, idx + 1, currentSum + numbers[idx]) + findWaysToMakeTarget(numbers, target, idx + 1, currentSum - numbers[idx]);
    }
}

다른 사람의 풀이

import java.util.LinkedList;
import java.util.Queue;

class Solution {
    Queue<Integer> q = new LinkedList<Integer>();
    public int solution(int[] numbers, int target) {
        int answer = 0;
        q.offer(0);
        q.offer(0);
        while(q.peek()!=null){
            int level =q.poll();
            int v = q.poll();
            if(level==numbers.length){
                if(v==target)
                    answer++;
            }else {

                q.offer(level + 1);
                q.offer(v + numbers[level]);

                q.offer(level + 1);
                q.offer(v - numbers[level]);
            }
        }
        return answer;
    }   
}
  • 숫자들을 더하거나 빼는 모든 경우의 수를 구해야 해서 DFS만 생각했었는데 BFS를 사용해서 푼 풀이가 있어 기록한다
  • 레벨과, 누적 합을 같은 queue에다 담아서 계산한 것이 신기했다 😛
profile
의견 나누는 것을 좋아합니다 ლ(・ヮ・ლ)

0개의 댓글