NYOJ-1058部分和问题

描述

给定整数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 <iostream>
#include <cstdio>
#include <cstring> 
using namespace std;
int num[22];
bool numisused[22];
bool flag ;
int N,K;
void dfs(int n,int sum)//n表示当前的第几个数字，sum表示当前的总和 
{
	if(flag==true)//这个标记表明我们已经找到我们需要的答案了，跳出多余的循环 
		return ;
	int i;
	if(n==N || sum<=0)//递归出口 ,当sum<=0的时候我们就退出，但是 只有sum==0的时候我们才输出
	{
		if(sum==0)
		{
			cout<<"YES"<<endl;
			for(i=0;i<N;i++)
			{
				if(numisused[i]==true)
					cout<<num[i]<<" ";
			}
			cout<<endl;
			flag = true;
		}
		return;
	}

	numisused[n] = true;//这里的三行表示我们尝试当前值 
	dfs(n+1,sum-num[n]);
	numisused[n] = false;	

	if(flag!=true) //这里也是一个小小的剪枝，当上面的退出循环已经找到答案了以后，我们就没必要看后面的了，和函数开头的那个if道理一样
	dfs(n+1,sum);//这一步非常关键，它表示我们需要跳过当前值， 
	
}

int main()
{
	
	while(cin>>N>>K)
	{
		flag = false;//初始化标记 
		for(int i=0;i<N;i++)
			cin>>num[i];
		memset(numisused,0,sizeof(numisused));	//初始化标记数组 
		dfs(0,K); 
		if(flag == false)
			cout<<"NO"<<endl;
	}
	return 0;
}

优化后时间的对比，

下面是纯代码

#include <iostream>
#include <cstdio>
#include <cstring> 
using namespace std;
int num[22];
bool numisused[22];
bool flag ;
int N,K;
void dfs(int n,int sum)
{
	if(flag==true) 
		return ;
	int i;
	if(n==N || sum<=0)
	{
		if(sum==0)
		{
			cout<<"YES"<<endl;
			for(i=0;i<N;i++)
			{
				if(numisused[i]==true)
					cout<<num[i]<<" ";
			}
			cout<<endl;
			flag = true;
		}
		return;
	}
	
		numisused[n] = true;
		dfs(n+1,sum-num[n]);
		numisused[n] = false;
		
	if(flag!=true)	
		dfs(n+1,sum);	
}

int main()
{
	
	while(cin>>N>>K)
	{
		flag = false;
		for(int i=0;i<N;i++)
			cin>>num[i];
		memset(numisused,0,sizeof(numisused));	
		dfs(0,K); 
		if(flag == false)
			cout<<"NO"<<endl;
	}
	return 0;
}

奈何我冒泡的算法如何打动你超时的心！！