部分和问题

部分和问题 时间限制：1000 ms | 内存限制：65535 KB
难度：3
描述:
给定整数a1、a2、.......an，判断是否可以从中选出若干数，使它们的和恰好为K。
输入:
首先，n和k，n表示数的个数，k表示数的和。
接着一行n个数。
（1<=n<=20,保证不超int范围）
输出:
如果和恰好可以为k，输出“YES”，并按输入顺序依次输出是由哪几个数的和组成，否则“NO”
样例输入:
5 11
1 2 3 5 7
样例输出:
YES
1 3 7
1 2 3 5

咳咳，部分和问题算是深度优先搜索的经典例子了，还记得第一次接触这类问题的时候，只是确定是否存在一组数据能与 k 相等，就是只要输出“ YES” 或者 “NO”就行了；

#include<stdio.h>
#include<iostream>
#include<algorithm>
using namespace std;
int a[100],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]))
	{
		return true;
	}
	return false;
}
int main()
{
	cin >> n >> k;
	for(int i=0;i<n;i++)cin >> a[i];
	if(dfs(0,0))cout << "YES" << endl;
	else  cout << "NO" << endl;
	return 0;
}

后来想将数给输出来，看看都有啥方案，越想越烦，越烦越想搞，然后诸多改进后完成了以下代码。。

#include<stdio.h>
#include<iostream>
#include<algorithm>
using namespace std;
int a[100],n,k;
bool judge[100],flag=false;
void display(int a[])			//对已确定的数组进行输出
{
	for(int i=0;i<n;i++)
		if(judge[i])cout << a[i] << " "; 
	cout << endl;
}
bool dfs(int i,int sum)			//利用深度优先搜索
{
	if(i==n)return sum==k;
	judge[i]=0;
	if(dfs(i+1,sum))
	{
		if(!flag)cout << "YES" << endl;
		display(a);
		flag=true;
	}
	judge[i]=1;
	if(dfs(i+1,sum+a[i]))
	{
		if(!flag)cout << "YES" << endl;
		display(a);
		flag=true;
	}
	return false;
}
int main()
{
	cin >> n >> k;
	for(int i=0;i<n;i++)cin >> a[i];
	dfs(0,0);
	if(!flag)cout << "NO" << endl;
	return 0;
}