취미처럼

[백준] 1182번 부분수열의 합 본문

Algorithm/백준

[백준] 1182번 부분수열의 합

sirius 2021. 3. 4. 17:26
https://www.acmicpc.net/problem/1182
 

1182번: 부분수열의 합

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

www.acmicpc.net

 

  • 다른 dfs와 달리 for문을 사용할 필요 없음
  • 해당 원소를 포함한 부분수열의 합해당 원소를 포함하지 않은 부분수열의 합을 따로 구해야 함
  • 정수의 합이 0일 때는 공집합도 포함하므로 count에서 1을 빼준다.

 

 

import java.util.*;
import java.io.*;

public class Main {

    public static int n;
    public static int num;
    public static int[] arr;
    public static int count = 0;

    public static void main(String[] args) throws Exception {

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

        dfs(0,0);

        if(num == 0) {
        	System.out.println(count-1);
        } else {
        	System.out.println(count);
        }
    }

    public static void dfs(int depth, int sum) {
        if(depth == n) {
            if(num == sum) {
            	count++;
            }
            return;
        }
        dfs(depth + 1, sum + arr[depth]);
        dfs(depth + 1, sum);
    }

}

 

 

'Algorithm > 백준' 카테고리의 다른 글

[백준] 1463번 1로 만들기  (0) 2021.03.05
[백준] 14391번 종이조각  (0) 2021.03.04
[백준] 11723번 집합  (0) 2021.03.04
[백준] 6603번 로또  (0) 2021.03.04
[백준] 10971번 외판원 순회 2  (0) 2021.03.04
Comments