1. 문제
N개의 정수로 이루어진 수열이 있을 때, 크기가 양수인 부분수열 중에서 그 수열의 원소를 다 더한 값이 S가 되는 경우의 수를 구하는 프로그램을 작성하시오.
2. 입력
첫째 줄에 정수의 개수를 나타내는 N과 정수 S가 주어진다. (1 ≤ N ≤ 20, |S| ≤ 1,000,000) 둘째 줄에 N개의 정수가 빈 칸을 사이에 두고 주어진다. 주어지는 정수의 절댓값은 100,000을 넘지 않는다.
3. 출력
첫째 줄에 합이 S가 되는 부분수열의 개수를 출력한다.
dfs(depth + 1, sum + num[depth]);
dfs(depth + 1, sum);
위와 같은 트리의 전위순회 구조로 작동합니다.
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);
}
}
이 문제 때문에 스트레스 많이 받았는데 덕분에 풀었습니다! 좋은 풀이 감사합니다