部分和问题【DFS】

题目来源：
挑战程序设计竞赛p30。

题目描述
给定整数a1、a2、…an，判断是否可以从中选出若干数，使它们的和恰好为k。

输入
首先输入n和k，n表示数的个数，k表示数的和。
接着一行n个数。（1<=n<=20,保证不超int范围）

输出
如果和恰好可以为k，输出“YES”，否则“NO”。

样例输入
4 13
1 2 4 7

样例输出
YES

程序说明：

深度优先搜索，每个数加或不加有两种状态。

代码如下：

#include <iostream>
#define MAX 100
using namespace std;
int n, k;
int a[MAX];
//已经从前i项得到了和sum，然后对于i项之后的进行分支
bool dfs(int i, int sum) {
	//如果前n项都计算过了，则返回sum与k是否相等
	if(i == n)
		return sum == k ? true : false;
	//不加上a[i]的情况
	if(dfs(i + 1, sum)) return true;
	//加上a[i]的情况
	if(dfs(i + 1, sum + a[i])) return true;
	//无论是否加上a[i]都不和k相等则返回false
	return false;
}
int main() {
	cin>>n>>k;
	for(int i = 0; i < n; i++)
		cin>>a[i];
	if(dfs(0, 0))
		cout<<"YES";
	else
		cout<<"NO";
	return 0;
}