问题描述
子集和问题的一个实例为<S,c>。其中S={x1,x2,…,xn}是一个正整数的集合,c是一个正整数。子集和问题判定是否存在S的一个子集S1,使得S1中所有元素的和为c
试设计一个解子集和问题的回溯法
分析
对于每一个集合中的元素,有取和不取两种状态,可以对这两种状态进行回溯,直到找完所有情况
用i记录回溯的当前元素索引号,v保存当前子集的和
递归出口 i>n 或 v>c
代码
#include <iostream>
using namespace std;
int n; //集合大小
int *s; //集合
int c; //目标值
int v; //当前值
bool *visit; //标记是否添加到子集
void backtrack(int i)
{
if (i > n || v > c)
{
return; //递归出口
}
else if (v == c)
{
//符合条件
for (int j = 0; j < n; j++)
{
if (visit[j])
{
cout << s[j] << ' ';
}
}
cout << endl;
}
else
{
v += s[i];
visit[i] = true; //取第i个加入子集
backtrack(i + 1);
v -= s[i];
visit[i] = false; //第i个不加入子集
backtrack(i + 1);
}
}
int main()
{
cin >> n >> c;
s = new int[n];
visit = new bool[n];
for (int i = 0; i < n; i++)
{
cin >> s[i];
visit[i] = false;
}
backtrack(0);
delete[] s;
delete[] visit;
return 0;
}
输入
6 15
4 5 6 7 8 10
输出
4 5 6
5 10
7 8