nyoj-部分和问题

部分和问题

时间限制： 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

思路：

    每一个数都加起来，如果使s=sum即找到了结果，否则，取消该数的标记，继续试下一个。

    当全都搜索一遍之后f仍然等于0，则未找到结果输出no

代码：

       

#include<stdio.h>
#include<string.h>
int a[25],book[25];
int n,sum,f,s;
void dfs(int step)
{
	if(s>=sum)
	{
		if(s==sum)//结束条件 
		{
			if(f==0)//如果找到就标记为1 
			  f=1;
			if(f)
			{
				printf("YES\n");
				for(int i=0;i<n;i++)
				{
					if(book[i])
					{
						printf("%d ",a[i]);
					}
				}
				printf("\n");
			}
		}
	}
	for(int k=step;k<n;k++)//每一个数都叠加 
	{
		s+=a[k];
		book[k]=1;//把用过的数标记为1 
		dfs(k+1);//搜索下一个 
		s-=a[k];//一旦超过sum就把s减去当前的a[k] 
		book[k]=0;//并把book[k]取消标记 
	}
}
int main()
{
	while(scanf("%d%d",&n,&sum)!=EOF)
	{
		f=0,s=0;
		memset(book,0,sizeof(book));
		for(int i=0;i<n;i++)
		{
			scanf("%d",&a[i]);
		}
		dfs(0);
		if(f==0)
		{
			printf("NO\n");
		}
	}
	return 0;
}