子集和问题——回溯法

问题描述

子集和问题的一个实例为<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