算法设计-子集和问题

子集和问题
Time Limit: 1000 ms Memory Limit: 65536 KiB

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<bits/stdc++.h>
#define INF 1000000
using namespace std;
int n, c;
int s[10005];
int ans[10006];
int k = 0;
int flag = 0;

void DFS(int i, int sum){
    if(sum > c){ //剪枝
        return;
    }
    if(sum == c){ //终点
        flag = 1;
        return;
    }
    for(int j = i+1; j <= n; j++){ //循环
        if(flag == 1) return; //剪枝
        ans[k++] = s[j]; //添加状态
        DFS(j, sum+=s[j]); //DFS
        if(!flag){ //没有找到则回溯状态
            k--;
            sum -= s[j];
        }
    }
}

int main()
{
    scanf("%d %d", &n, &c);
    int ss = 0;
    for(int i = 1; i <= n; i++){
        scanf("%d", &s[i]);
        ss+=s[i];
    }
    if(ss < c){
        printf("No Solution!");
        return 0;
    }
    s[0] = 0;
    DFS(0, 0);
    if(!flag){
        printf("No Solution!");
        return 0;
    }
    for(int i = 0; i < k; i++){
        printf("%d", ans[i]);
        if(i != k-1){
            printf(" ");
        }else{
            printf("\n");
        }
    }
}