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

쓰리원·2022년 3월 1일
2

코딩테스트

목록 보기
7/28
post-thumbnail
post-custom-banner

1. 백준 1182번 문제

1. 문제

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

2. 입력

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

3. 출력

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

2. 문제 해석

dfs(depth + 1, sum + num[depth]);
dfs(depth + 1, sum);

위와 같은 트리의 전위순회 구조로 작동합니다.

3. 문제 풀이

import java.util.Scanner;

class Main {
    static int[] num;
    private static int N; // 정수의 개수
    private static int S; // 정수의 합
    private static int answer = 0;

    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        N = sc.nextInt();
        S = sc.nextInt();
        num = new int[N];
        for (int i = 0; i < N; i++) {
            num[i] = sc.nextInt();
        }
        dfs(0, 0);
        if (S == 0) System.out.println(answer - 1); 
        //S 0일때가 전체 합이 0 일때랑 겹치기 때문에 -1 을 해주어야 합니다.
        //위의 사진에 공집합이 0이라고 볼 수 있습니다
         else  System.out.println(answer);

    }

    private static void dfs(int depth, int sum) {
        if (depth == N) {
            if (sum == S) answer++;
            return;
        }
        //tree를 생각해서 직접 그려보면 이해하기가 쉽습니다.
        dfs(depth + 1, sum + num[depth]);
        dfs(depth + 1, sum); 
    }
}
profile
가장 아름다운 정답은 서로의 협업안에 있다.
post-custom-banner

1개의 댓글

comment-user-thumbnail
2023년 8월 30일

이 문제 때문에 스트레스 많이 받았는데 덕분에 풀었습니다! 좋은 풀이 감사합니다

답글 달기