子集和问题

转载于：http://blog.csdn.net/clove_unique
　　　　　子集和问题
【问题描述】
　　　　子集和问题的一个实例为〈S,t〉。其中，S={ x1， x2，…， xn}是一个正整数的集合，c是一个正整数。子集和问题判定是否存在S的一个子集S1，使得子集S1和等于c。
【编程任务】
　　　　对于给定的正整数的集合S={ x1， x2，…， xn}和正整数c，编程计算S 的一个子集S1，使得子集S1和等于c。
【输入格式】
　　　　第1行有2个正整数n和c，n表示S的个数，c是子集和的目标值。接下来的1 行中，有n个正整数，表示集合S中的元素。
【输出格式】
　　　　程序运行结束时，将子集和问题的解输出。当问题无解时，输出“No solution!”。
【输入样例】
　5 10
　2 2 6 5 4
【输出样例】
　2 2 6

【解题思路】
水题~~~与前面异曲同工。。。
结束的判断条件不同，之前是数据的个数达到要求就停止，而现在是数据之和等于c停止。。而且只需输出一组，直接停止即可。。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cstdlib>
using namespace std;
int n,c,a[10000],b[10000],d[10000],i,sum=0;
bool pd;

void print(int dep)
{
    int j;
    for (j=1;j<=dep;++j)
      printf("%d ",d[j]);
    printf("\n");
    return;
}

void dfs(int dep)
{
    int r;
    if (pd==1) return;//因为只需输出一组解，c++又没法结束整个程序，，怕超时到处加判断。。。（我智商低别理我）
    for (r=1;r<=n;++r)
      if (!b[r]&&c-a[r]>=0)
      {
        if (pd==1) return;
        c-=a[r];//用c减去每个数，其实就是几个数之和等于c
        d[dep]=a[r];//d是记录数组
        b[r]=1;
        if (c==0) //如果找到一组解，直接输出，完成任务
        {
            if (pd==0)
              print(dep);
            pd=1;
            return;
        }
        else dfs(dep+1);
        if (pd==1) return; 
        c+=a[r];//回溯一步
        b[r]=0;
      }
}

int main()
{
    scanf("%d%d",&n,&c);
    for (i=1;i<=n;++i)
    {
        scanf("%d",&a[i]);
        sum+=a[i];
    }
    if (sum<c)//很重要的剪枝！！如果所有的数加起来都小于c，那么不可能有解。。之前有三组TLE，加了这一步竟然给蒙过了。。
    {
        printf("No solution!");
        return 0;
    }
    memset(d,0,sizeof(d));
    memset(b,0,sizeof(b));
    pd=0;
    dfs(1);
    if (!pd) printf("No solution!");
    return 0;
}