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;
}