NYOJ-部分和问题（dfs）

                                         部分和问题

描述：

    给定整数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

解题思路：

先从第一个数开始搜，并把sum初始化为0，然后分两种情况一种是把下一个数加上并标记，另一种是不把下一个数加上并标记

程序代码：

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