NYOJ - 1058 - 部分和问题(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

题目思路：拿到这道题，看到从给定数列中选出若干个数，是它们的和恰好为K的时候，首先想到子集，用子集生成的递归算法，求出所有子集，判断符合条件的子集并且输出。后来想了半天，不知道怎么写。于是改变了思路。用深度搜索，从左往右，挨个元素搜索，但是递归的时候只搜索当前元素右边的元素。(其实就是求全排列的方法，只不过不搜左边，因为相同数字，顺序不同则是不同的排列，但却是相同的子集)。递归的边界是sum>=k，搜到以后直接停止搜索。

#include<cstdio>
#include<iostream>
#include<string>
#include<cstring>
#include<cmath>
#include<algorithm>
using namespace std;
int n,k;
int sum;
int a[22];
int vis[22];
int b;
void dfs(int cur){
if(sum>=k){
if(sum==k){
b = 1;//用来判断是否存在有效输出
puts("YES");
for(int i=0 ;i<n ;i++){
if(vis[i])
printf("%d ",a[i]);
}
puts("");

}
return ;
}
for(int i=cur ;i<n ;i++){
sum += a[i];
vis[i] = 1;//设计使用标记
dfs(i+1);//这里的变量是i+1 不是cur+1 因为不搜索左边
sum -= a[i];
vis[i] = 0; //清除使用标记
}

}

int main(){
while(scanf("%d%d",&n,&k)!=EOF){
sum = 0;
b=0;
memset(vis,0,sizeof(vis));
for(int i=0 ;i<n ;i++){
scanf("%d",&a[i]);
}
dfs(0);
if(!b)
puts("NO");
}

return 0;
}