描述
子集和问题的一个实例为〈S,t〉。其中,S={ x_1, x_2,…, x_n}S=x1,x2,…,xn是一个正整数的集合,c是一个正整数。子集和问题判定是否存在S的一个子集S1,使得子集S1和等于c
对于给定的正整数的集合S={ x_1, x_2,…, x_n}S=x1,x2,…,xn和正整数c,编程计算S 的一个子集S1,使得子集S1和等于c。
输入
第1行有2个正整数n和c,n表示S的个数,c是子集和的目标值。接下来的1 行中,有n个正整数,表示集合S中的元素。
2<=n<=20002<=n<=2000
输出
将子集和问题的解输出。当问题无解时,输出“No Solution!”。
#include<bits/stdc++.h>
using namespace std;
const int N = 1e3 + 10;
int a[N];//用于存储S的正整数集合
bool x[N];//用于记录各分支的结果情况
int tw = 0,rw = 0;//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)//打印这条路径上的每个元素(用x[] = 1标记)
cout << a[j] << " ";
flag = true;
return ;
}
if(tw + a[i] <= c){//小于 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);//从第 0 个结点开始递归遍历
if(flag == false)
cout << "No Solution!\n";
return 0;
}