【回溯算法】子集和问题

算法实现题 5-1 子集和问题

【问题描述】
　子集和问题的一个实例为〈S,t〉。其中，S={ x1， x2，…， xn}是一个正整数的集合，c是一个正整数。子集和问题判定是否存在S的一个子集S1，使得子集S1和等于c。
【编程任务】
　　对于给定的正整数的集合S={ x1， x2，…， xn}和正整数c，编程计算S 的一个子集S1，使得子集S1和等于c。
【输入格式】
　　由文件subsum.in提供输入数据。文件第1行有2个正整数n和c，n表示S的个数，c是子集和的目标值。接下来的1 行中，有n个正整数，表示集合S中的元素。
【输出格式】
程序运行结束时，将子集和问题的解输出到文件subsum.out中。当问题无解时，输出“No Solution!”。
【输入样例】
5 10
2 2 6 5 4
【输出样例】
2 2 6
 

参考文章：https://blog.csdn.net/qq_28640193/article/details/90178416 无剪枝方法。

https://blog.csdn.net/gl620321/article/details/108801724 以下内容摘自该博客

求解该问题需要搜索整个解空间树，设解向量x=（x1，x2，…,xn）,本问题是求所有解，所以一旦搜索到叶子结点（即i=n+1）。如果相应的子集和为W，则输出x解向量。搜索到第i（1<=i<=n）层的某个结点时用tw表示选取的整数和，rw表示余下的整数和，rw表示余下的整数和，rw=w[j] (j从i+1到n)

（1）约束函数：检查当前整数w[i]加入子集和是否超过W，若超过，则不能选择该路径。用于左孩子结点剪枝。

（2）限界函数：一个结点满足tw+rw<W,即即使选择剩余的所有整数，也不可能找到一个解。用于右孩子剪枝。
 

按照如下代码，一般都会先考虑树的左边，如果左边全部都不符合条件的话，回溯返回，接着遍历右子树。

有剪枝

#include<bits/stdc++.h>

using namespace std;
//含有剪枝
int a[999];//用于存储S的正整数集合
bool x[999];//用于记录各分支的结果情况
int tw=0,rw;//tw：当前整数和，rw：余下的整数和
bool flag=false;//记录是否有解
int n,c;

void backtrack(int i) {
	if(i>n||flag==true)
		return ;
	if(tw==c) {
		for(int j=0; j<i; j++)
			if(x[j]==1)
				cout<<a[j]<<" ";
		flag=true;
		return ;
	}
	if(tw+a[i]<=c) {//进入左子树,剪枝
		x[i]=1;
		tw=tw+a[i];
		rw=rw-a[i];
		backtrack(i+1);
		tw=tw-a[i];
		rw=rw+a[i];
	}
	if(tw+rw-a[i]>=c) {//进入右子树，剪枝
		x[i]=0;
		rw=rw-a[i];
		backtrack(i+1);
		rw=rw+a[i];
	}
	return ;
}

int main() {
	cin>>n>>c;
	for(int i=0; i<n; i++) {
		cin>>a[i];
		rw=rw+a[i];
	}
	backtrack(0);
	if(flag==false)
		cout<<"No Solution!\n";
	return 0;
}
/*
5 10
2 2 6 5 4
2 2 6
*/

 

无剪枝

#include<bits/stdc++.h>

using namespace std;

int a[999];//用于存储S的正整数集合
bool x[999];//用于记录各分支的结果情况  
int w=0;//用于记录此时的子集和
bool flag=false;//记录是否有解
int n,c;

void backtrack(int i) {
	if(i>n||flag==true)//到了最后一层或者已找到子集 
		return ;
	//取出该层的数字
	x[i]=true;
	w=w+a[i];
	//判断情况
	if(w==c) {
		for(int j=0; j<=i; j++) {
			if(x[j]==true)
				cout<<a[j]<<" ";
		}
		flag=true;
		return ;
	} else if(w<c) {
		//接着向下取数
		backtrack(i+1);
	}
	//回溯算法
	x[i]=false;
	w=w-a[i];
	backtrack(i+1);
	return ;
}

int main() {
	cin>>n>>c;
	for(int i=0; i<n; i++)
		cin>>a[i];
	backtrack(0);
	if(flag==false)
		cout<<"No Solution!\n";
	return 0;
}
/*
5 10
2 2 6 5 4
2 2 6
*/

下面的这个代码，可以找到所有的符合条件的子集。

而且按照我的这种判断，相等的数字需要被判断两次，所以也就会出现相同的结果。

这个属于排序树。

#include<bits/stdc++.h>

using namespace std;

int a[999];//用于存储S的正整数集合
bool x[999];//用于记录各分支的情况
int w=0;//用于记录此时的子集和
int n,c;

void backtrack(int i) {
	if(i>n) {
		return ;
	}
	//取出该层的数字
	x[i]=true;
	w=w+a[i];
	//判断情况
	if(w==c) {
		for(int j=0; j<=i; j++) {
			if(x[j]==true)
				cout<<a[j]<<" ";
		}
		cout<<endl;
	} else if(w<c) {
		//接着向下取数
		backtrack(i+1);
	}
	//回溯算法
	x[i]=false;
	w=w-a[i];
	backtrack(i+1);
	return ;
}

int main() {
	cin>>n>>c;
	for(int i=0; i<n; i++)
		cin>>a[i];
	backtrack(0);
	return 0;
}