部分和问题

问题链接：http://acm.nyist.net/JudgeOnline/problem.php?pid=1058

部分和问题

时间限制：1000 ms | 内存限制：65535 KB
难度：2
描述
给定整数a1、a2、…….an，判断是否可以从中选出若干数，使它们的和恰好为K。
输入
首先，n和k，n表示数的个数，k表示数的和。
接着一行n个数。
（1<=n<=20,保证不超int范围）
输出
如果和恰好可以为k，输出“YES”，并按输入顺序依次输出是由哪几个数的和组成，否则“NO”
样例输入
4 13
1 2 4 7
样例输出
YES
2 4 7

分析：

一共输入n个数，每个数有两种状态，分别为加入解集合、不加入解集合。
每个数作为解空间树的一层，则树有n层。利用递归调用对树进行深度优先遍历，即可找到解。
因为n<=20，所以树空间大小最大为2^20，在百万级别以内。

代码：

#include <iostream>
#include <stdio.h>
#define MAX 21
/* run this program using the console pauser or add your own getch, system("pause") or input loop */
int num[MAX];
bool mark[MAX];
int n,k;
bool dfs(int i,int sum){
    //递归遍历到第i个数
    if(i == n) {  //每次都深度递归到i == n 然后判断进行返回 
        return sum == k;
    }

    //加上第i个数 
    mark[i] = false;
    if(dfs(i+1,sum)){
        return true;
    }

    //不加第i个数 
    mark[i] = true;
    if(dfs(i+1,sum+num[i])){
        return true;
    }

    //加上和不加最终都得不到sum == k 则返回false 
    return false;
}
int main(int argc, char** argv) {

    while(scanf("%d%d",&n,&k)!=EOF){
        //1.输入n个数  
        for(int i=0;i<n;i++){
            scanf("%d",&num[i]);
            mark[i] = false; 
        }
        //2.深度搜索dfs(i,j)表示遍历到第i个数，此时的和为j 
        if(dfs(0,0)){
            printf("YES\n") ;
            for(int i=0;i<n;i++){
                if(mark[i] == true){
                    printf("%d ",num[i]) ;
                }
            }
            printf("\n");
        }else{
            printf("NO\n");
        }
    }

    return 0;
}