一、内容
子集和问题的一个实例为<S,c>。其中,S={x1,x2,...,xn}是一个正整数的集合,c是一个正整数。子集和问题判定是否存在S的一个子集S1,使得
试设计一个解子集和问题的回溯法。
对于给定的正整数的集合S和正整数c,计算S的一个子集S1使得
输入
第1行有2个正整数n和c,n表示S的大小,c是子集和的目标值。接下来的1行中,有n个正整数,表示集合S中的元素。
输出
将子集和问题的解输出。当问题无解时,输出“Solution!”
样例输入 Copy
5 10
2 2 6 5 4
样例输出 Copy
2 2 6
二、思路
- 可行性剪枝:当我们组成的数的和sum > 我们要求解的值m时, 直接返回不再往下继续搜索。
- 重复性剪枝:我们从已经选了的数后面进行选取,这样就避免了重复的过程。 比如选择1 2 3, 若不从下个位置进行搜索,那么还会搜索出2 1 3, 2 3 1 这些重复的结果。
三、代码
#include <cstdio>
const int N = 10005;
int a[N], n, m, rec[N];
bool ok;
void dfs(int start, int sum, int cnt) {
if (sum > m) return ;//剪枝
if (sum == m) {
for (int i = 0; i < cnt; i++) printf("%d ", rec[i]);
ok = true;//记录是否找到一组解
return ;
}
for (int i = start; i <= n; i++) { //重复性剪枝
rec[cnt] = a[i]; //记录当前选的值
dfs(i + 1, sum + a[i], cnt + 1); //对下一个进行搜索
if (ok) return;
}
}
int main() {
scanf("%d%d", &n, &m);
for (int i = 1; i <= n; i++) {
scanf("%d", &a[i]);
}
dfs(1, 0, 0);
if (!ok) printf("Solution!");
return 0;
}