子集和问题

Problem Description

子集和问题的一个实例为〈S,t〉。其中，S={  x1 ， x2 ，…，xn }是一个正整数的集合，c是一个正整数。子集和问题判定是否存在S的一个子集S1，使得：
。
试设计一个解子集和问题的回溯法。
对于给定的正整数的集合S={  x1 ， x2 ，…，xn }和正整数c，计算S 的一个子集S1，使得：
。

Input

输入数据的第1 行有2 个正整数n 和c（n≤10000，c≤10000000），n 表示S 的大小，c是子集和的目标值。接下来的1 行中，有n个正整数，表示集合S中的元素。

Output

将子集和问题的解输出。当问题无解时，输出“No Solution!”。

Sample Input

5 10

2 2 6 5 4

Sample Output

2 2 6

#include <iostream>

using namespace std;

int n, c;
int s[10010], dp[10010];
int sum = 0, k = 0;

bool imag = false;

void DFS(int i) {//回溯法
    sum += s[i];
    dp[k++] = s[i];
    if (sum > c) return;
    if (sum == c) {
        imag = true;
        return;
    }
    for (int j = 1 + i; j <= n; j++) {
        if (imag) return;
        DFS(j);
        if (!imag) {
            sum -= s[j];
            k--;
        }
    }
}

int main() {
    cin >> n >> c;
    int ss = 0;
    for (int i = 1; i <= n; i++) {
        cin >> s[i];
        ss += s[i];
    }

    if (ss < c) {//体前判断，省时
        cout << "No Solution!";
        return 0;
    }

    s[0] = 0;//从头遍历，比较方便
    DFS(0);

    if (imag) {
        for (int i = 1; i < k; i++) {
            cout << dp[i];
            if (i < k - 1)
                cout << " ";
        }
    }
    return 0;
}