NYOJ-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

#include<cstdio>
#include<stack>

using namespace std;

stack<int> v;
const int maxn = 22;
int a[maxn];
int n , k ;

bool dfs(int i , int sum){
    if(i == n) return sum == k;
    if(sum > k) return false ;
    if(dfs(i+1 , sum)) return true ;
    if(dfs(i+1 , sum+a[i]) ){
        v.push(a[i]);
        return true;
    }
    return false;
}

int main(){
    while(~scanf("%d%d", &n , &k)){
        for(int i = 0 ; i< n ; i++){
            scanf("%d" , &a[i]);
        }
        if(dfs(0 , 0)){
            printf("YES\n");
            while(!v.empty()){
                printf("%d" , v.top());
                v.pop();
            }
        }
        else
            printf("NO");
    }
    return 0;
}