NYOJ-1058 部分和问题

题目部分：

部分和问题

时间限制： 1000 ms  |  内存限制： 65535 KB

难度： 2

描述

给定整数a1、a2、.......an，判断是否可以从中选出若干数，使它们的和恰好为K。

输入

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

输出

如果和恰好可以为k，输出“YES”，并按输入顺序依次输出是由哪几个数的和组成，否则“NO”

样例输入

4 13
1 2 4 7

样例输出

YES
2 4 7

深搜问题，递归实现

代码部分：

#include <stdio.h>
#include <string.h> 
int a[21];
bool b[21];
int n, k;
bool dfs(int i, int sum)
{
	if(i == n)
		return sum == k;
	if(dfs(i + 1, sum)) 
		return true;
	if(dfs(i + 1, sum + a[i])) 
	{
		b[i] = true;
		return true;
	}
	return false;
}
int main()
{
	int i;
	while(scanf("%d%d", &n, &k) != EOF)
	{
		for(i = 0; i < n; i++)
			scanf("%d", &a[i]);
		memset(b, 0, sizeof(b));
		if(dfs(0, 0))
		{
			printf("YES\n");
			for(i = 0; i < n; i++)
				if(b[i])
					printf("%d ", a[i]);
			printf("\n"); 
		}
		else
			printf("NO\n"); 
	}
	return 0;
}