본문 바로가기

알고리즘

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

반응형

출처 : programmers.co.kr/learn/courses/30/lessons/43165

 

코딩테스트 연습 - 타겟 넘버

n개의 음이 아닌 정수가 있습니다. 이 수를 적절히 더하거나 빼서 타겟 넘버를 만들려고 합니다. 예를 들어 [1, 1, 1, 1, 1]로 숫자 3을 만들려면 다음 다섯 방법을 쓸 수 있습니다. -1+1+1+1+1 = 3 +1-1+1+1+

programmers.co.kr

문제 설명

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

풀이

1. 부분 조합 활용해보기에 좋은 문제이다.

코드 _ s_idx 는 필요없다.

class Solution {
    
    static boolean[] visited ;
    static int count ;
    
    public int solution(int[] numbers, int target) {
        int answer = 0;
        count =0;
        visited = new boolean[ numbers.length ];
        // 1, 조합을 구하자.
        perm(0,0,target, numbers);
        answer = count;
        return answer;
    }
    private static void perm(int idx, int s_idx, int target, int[] numbers){
        
        // 기저조건
        if( idx == numbers.length){
            int sum =0;
            for( int i=0; i<idx; i++){
                int number = numbers[i];
                if( visited[i]){
                    number = -number;
                }
                sum += number;
            }            
            if ( sum == target) count++;
            
            return;
        }
        
        visited[idx] = true;
        perm( idx +1, s_idx+1, target, numbers);
        visited[idx] = false;
        perm( idx +1, s_idx, target, numbers);
        
        
    }
    
    
}
반응형

'알고리즘' 카테고리의 다른 글

[삼성 SW Expert Academy] 등산로 1949[JAVA]  (0) 2020.11.04
[백준] 스도쿠 [JAVA]  (0) 2020.11.04
[프로그래머스] 주식가격 [JAVA]  (0) 2020.10.31
[백준] 파티 1238 [JAVA]  (0) 2020.10.31
[백준] 사이클 게임 20040 [JAVA]  (0) 2020.10.30