티스토리 뷰

728x90

문제링크

문제

N개의 정수로 이루어진 수열이 있을 때, 크기가 양수인 부분수열 중에서 그 수열의 원소를 다 더한 값이 S가 되는 경우의 수를 구하는 프로그램을 작성하시오.

입력

첫째 줄에 정수의 개수를 나타내는 N과 정수 S가 주어진다. (1 ≤ N ≤ 20, |S| ≤ 1,000,000) 둘째 줄에 N개의 정수가 빈 칸을 사이에 두고 주어진다. 주어지는 정수의 절댓값은 100,000을 넘지 않는다.

출력

첫째 줄에 합이 S가 되는 부분수열의 개수를 출력한다.

 

 

문제 풀이 과정

배열의 모든 원소를 탐색하면서 원소를 선택 했을 때와 선택 하지 않을 때의 합이 target 이 되면 count 를 증가 시키고 그렇지 않으면 다음 탐색을 진행 하면 된다.

 

public static void dfs(int index, int sum){
    if(index == N){
        if(sum == target){
            count++;
        }
        return;
    }

    dfs(index + 1, sum + arr[index]);
    dfs(index + 1, sum);
}

 

dfs(index + 1, sum + arr[index]); 이 것이 원소를 선택 했을 때 합에다가 더해서 넘기는 것이고, dfs(index + 1, sum); 이것은 원소를 선택 하지 않았을 때의 경우이다.

 

System.out.println(target == 0 ? count - 1 : count);

 

마지막 결과를 출력 할때 0인 경우를 고려한 이유는 공집합도 될 수 있기 때문에 1개를 빼고 출력해야 한다.

 

 

전체코드

728x90