首页 > 其他 > 详细

深搜之搜索树

时间:2014-02-21 17:24:51      阅读:270      评论:0      收藏:0      [点我收藏+]
给定整数a1、a2、.......an,判断是否可以从中选出若干数,使它们的和恰好为K。
输入:
n=4
a={1,2,4,7}
k=13
输出:
Yes(13=2+4+7)
经典深搜问题:递归代码:
#include <cstdio>
int a[30];int n,k;
bool dfs(int i,int sum)  //前i项之和sum
{
    if(i==n)
        return sum==k;
    if(dfs(i+1,sum))  //放入第i个
        return true;
    if(dfs(i+1,sum+a[i]))  //不放入第i个
        return true;
}
int main()
{
    while(~scanf("%d%d",&n,&k))
    {
        for(int i=0;i<n;i++)
            scanf("%d",&a[i]);
        if(dfs(0,0))
            printf("YES\n");
        else
            printf("NO\n");
    }
    return 0;
}

如果要求保存相加的每一个数,可以这样实现:
#include <iostream>
#include <vector>
using namespace std;
int a[4]={1,2,4,7},k;
vector<int> ans;
void dfs(int i,int sum)
{
	if(i==4){
		if(sum==k)
	   	{
   			for(int i=0;i!=ans.size();i++)cout<<ans[i]<<" ";
			cout<<endl;
			return ;
   		}
   		else 
		 return ;
	}
	dfs(i+1,sum);
	ans.push_back(a[i]);
	dfs(i+1,sum+a[i]);
	ans.pop_back();
	return ;
}
int main()
{
	 k=7;
	 dfs(0,0);
}


深搜之搜索树

原文:http://blog.csdn.net/y990041769/article/details/19603075

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!