子集和问题(pta)-标准回溯法

子集和问题(pta)-标准回溯法

题目位置:

pta遇到的,其他网站还没有找到。


问题描述:

设集合S={x1,x2,…,xn}是一个正整数集合,c是一个正整数,子集和问题判定是否存在S的一个子集S1,使S1中的元素之和为c。试设计一个解子集和问题的回溯法,并输出利用回溯法在搜索树(按输入顺序建立)中找到的第一个解。

输入格式:输入数据第1行有2个正整数n和c,n表示S的大小,c是子集和的目标值。接下来的1行中,有n个正整数,表示集合S中的元素。
是子集和的目标值。接下来的1 行中,有n个正整数,表示集合S中的元素。

5 10
2 2 6 5 4


输出格式:输出利用回溯法找到的第一个解,以空格分隔,最后一个输出的后面有空格。当问题无解时,输出“No Solution!”。

2 2 6


原因分析与代码:

本题是一道回溯题,是一道很经典的子集树类型问题,就需要对每个位置是否选择做树的判断,要设置上界函数进行剪枝,这样可以构成优化。

#include<bits/stdc++.h>
using namespace std;
const int N=1e5+10;
int n,m,a[N],r=0,x[N]={0},sum;
bool dfs(int i)
{
	if(i>=n){
		if(sum==m) return true;
		else return false;
	}
	else
	{
		r-=a[i];
		if(sum+a[i]<=m)
		{
			x[i]=1;
			sum+=a[i];
			if(dfs(i+1)) return true;
			sum-=a[i];
		}
		if(sum+r>=m)
		{
			x[i]=0;
			if(dfs(i+1)) return true;
		}
		r+=a[i];
		return false;
	}
}
int main()
{
	cin>>n>>m;
	for(int i=0;i<n;i++)
	{
		cin>>a[i];
		r+=a[i];
	}
	if(dfs(0))
	{
		for(int i=0;i<n;i++) if(a[i] * x[i]!=0) cout<<a[i]*x[i]<<" ";
	}
	else cout<<"No Solution!"<<endl; 
}

总结:

本题是一道标准的回溯板子题,有学会合理运用上界函数,可以学习其中利用数组做树的方法,利用想成确定是否存在,是个很有效的方法。

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值