1182 부분수열의 합 (Java)

부분수열의 합

문제

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

입력

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

출력

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

풀이

모든 부분수열에서 그 원소들의 합이 S인 경우를 찾는 문제이다.
모든 경우의 수? => 백트레킹이다.
백트레킹을 돌며 부분수열에 원소가 추가될 때마다 sum을 구하고 그때 s가 맞는지를 비교한다.
sum을 구하는 방식은 백트레킹에 들어갈 때, 기존 백트레킹으로 얻은 부분수열의 합을 가지고 와서 다음 들어갈 수를 더하면 된다.

코드

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.StringTokenizer;

public class Main{

    static int ans = 0;

    static int s;

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st = new StringTokenizer(br.readLine());

        int n = Integer.parseInt(st.nextToken());
        s = Integer.parseInt(st.nextToken());

        int[] arr = new int[n];
        st = new StringTokenizer(br.readLine());
        for (int i = 0; i < n; i++) {
            arr[i] = Integer.parseInt(st.nextToken());
        }

        backTracking(arr, 0, 0, 0);
        System.out.println(ans);

    }

    static void backTracking(int[] arr, int depth, int sum, int idx) {
        if (depth == arr.length) {
            return;
        }

        for (int i = idx; i < arr.length; i++) {
            sum += arr[i];
            if(sum == s)
                ans++;
            backTracking(arr, depth + 1, sum, i + 1);
            sum -= arr[i];
        }

    }
}