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>
void search(int k,int ans);
int a[25];
bool b[25];
int flag; 
int n,k;
int main()
{
	while(scanf("%d%d",&n,&k)!=EOF)
	{
		flag=1;
		memset(b,0,sizeof(b)); 
		for(int i=0;i<n;i++)
			scanf("%d",&a[i]);
		search(0,0);
		if(flag)          //没有找到答案就输出NO，我原先想把search函数写成bool类型的，
			printf("NO\n");	//后来发现用标记更简单，以后要多学习学习 
	}
	return 0;
}
void search(int num,int ans)
{
	if(!flag)
		return;
	if(ans==k)
	{
		if(num!=0)
		{
			flag=0;
			printf("YES\n");
			for(int i=0;i<n;i++)
			{
				if(b[i])
				{
					printf("%d ",a[i]);
				}
			}
			printf("\n");
			return ;
		}
	}	
	for(int i=num;i<n;i++)
	{
		if(ans+a[i]<=k)
		{
			b[i]=1;
			search(i+1,ans+a[i]);
			b[i]=0;
		}
	}
	return ;
}