NYOJ-部分和问题（DFS基础）

**

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

AC代码。

#include <iostream>
#include <cstring>
using namespace std;
int a[25];
int flag;
int n,k,s;
int judge[25];
void dfs(int sum)//关于参数问题，往往将其设为变化中的量 
{
 	if(sum==k)//判定结束条件 
	 {
 		 flag=1;//存在结果输出 
 		 cout<<"YES"<<endl;
 		 for(int i=0 ; i<n ; i++)
 		 {
  			 if(judge[i]==1)//成功存入sum的数组成员方满足该条件。 
	 		 {
		    		cout<<a[i]<<" ";
  			 }
		 }
 	 cout<<endl;
 	 return ; 
 }
 for(int i=s;i<n;i++)//i以s为起点可以过滤大量不必要的询问
 {
  	if(judge[i]==0&&sum+a[i]<=k)//sum+a[i]<=k是一个小小的优化，可以节约700ms时间
 	 {
  		 sum+=a[i];
  		 judge[i]=1;
  		 s=i; 
  		 dfs(sum);
		 sum=sum-a[i];//  
  		 judge[i]=0;//  回溯 
 	 }
 }
 	return ;
}
int main()
{
	 while(cin>>n>>k)
	 {//初始化 
 		 flag=0;
		 s=0;
		 memset(a,0,sizeof(a));
 		 for(int i=0;i<n;i++)
 		 cin>>a[i];
  		 dfs(0);
 		 if(flag==0)//无结果输出 
 		 {
 		  cout<<"NO"<<endl;
 		 }
	 }
	 return 0;
}