部分和问题

部分和问题

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


下面是递归和回溯

#include <stdio.h>
#include <string.h>
int n,k,ok,arr[25],vis[25],count;
void DFS(int pos)
{
	if(count>=k)
	{
		if(count==k)
		{
			if(!ok)
			{
				ok=1;
				printf("YES\n");
			}
			for(int i=0;i<n;i++)
			{
				if(vis[i])
				{
					printf("%d ",arr[i]);
				}
			}
			printf("\n");
		}
		return;
	}
	for(int i=pos;i<n;i++)
	{
		count+=arr[i];
		vis[i]=1;
		DFS(i+1);
		count-=arr[i];
		vis[i]=0;
	}
}
int main()
{
	while(scanf("%d%d",&n,&k)!=EOF)
	{
		memset(vis,0,sizeof(vis));
		ok=0;
	   for(int i=0;i<n;i++)
	    {
		    scanf("%d",&arr[i]);
	   }
		count=0;
		DFS(0);
		if(!ok)
		{
			printf("NO\n");
		}
	}
	return 0;
}