[SW Expert Academy] - (D3)2817. 부분 수열의 합

https://swexpertacademy.com/main/code/problem/problemDetail.do?contestProbId=AV7IzvG6EksDFAXB&categoryId=AV7IzvG6EksDFAXB&categoryType=CODE

SW Expert Academy

코드

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

public class Solution {
	/*
	 * N:     자연수 갯수
	 * K:     자연수를 선택해서 합이 K가 되는 경우
	 * arr[]: N개의 자연수를 담을 배열
	 * ans:   경우의 수
	 */
	private static int N;
	private static int K;
	private static int arr[];
	private static int ans;
	
	private static void dfs(int count, int sum) {
		/* 기저사례: 자연수 x개 선택해서 합이 K가 되는 경우 */
		if(sum == K) {
			ans ++;
			return;
		}
		
		/* 기저사례: 자연수 x개의 합이 K보다 큰경우 || 선택한 자연수의 갯수가 N개와 동일한 경우*/
		if(sum > K || count == N) {
			return;
		}
		
		// 자기 자신 포함
		dfs(count+1, sum+arr[count]);
		
		// 자기 자신 미포함
		dfs(count+1, sum);
	}

	public static void main(String[] args) throws NumberFormatException, IOException {
		BufferedReader bf = new BufferedReader(new InputStreamReader(System.in));
		StringTokenizer st;
		StringBuilder sb = new StringBuilder();
		
		int t = Integer.parseInt(bf.readLine());
		
		for(int tc=1; tc<=t; tc++) {
			st = new StringTokenizer(bf.readLine());
			N = Integer.parseInt(st.nextToken());
			K = Integer.parseInt(st.nextToken());
			arr = new int[N];
			ans = 0;
			
			st = new StringTokenizer(bf.readLine());
			for(int i=0; i<N; i++) {
				arr[i] = Integer.parseInt(st.nextToken());
			}
			
			dfs(0, 0);
			System.out.println("#" + tc + " " + ans);
		}
		
		bf.close();
	}

}

풀이

nCr 조합 문제.

재귀함수를 통해 해결하고, 재귀함수의 종료 조건은 다음과 같다.

 - 고른 자연수의 합이 K와 동일한 경우

 - 고른 자연수의 합이 K보다 크거나, 고른 자연수의 갯수가 N개일 경우

 

dfs 재귀 호출을 통해 현재 선택된 숫자를 포함하는 경우

 - dfs(count+1, sum+arr[count])

 

현재 선택된 숫자를 포함하지 않고 다음으로 가는 경우

 - dfs(count+1, sum)

 

로 나누어 탐색을 진행함.