[DFS] 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

这题和NY927基本一样,只多了输出组成的数.

//已AC代码
#include<cstdio>
#include<stack>
#include<iostream>
using namespace std;
int n;
int a[1000];
int aim;
int sum;
stack<int> s;
bool dfs(int pos,int sum)
{
    if(pos==n)
    {
        return sum==aim;
    }
    if(dfs(pos+1,sum))
    {
        return true;
    }
    if(dfs(pos+1,sum+a[pos]))
    {
        s.push(a[pos]);
        return true;
    }
    return false;
}
int main()
{

    while(~scanf("%d",&n))
    {
        scanf("%d",&aim);
        for(int i=0;i<n;i++)
        {
            scanf("%d",&a[i]);

        }
        if(dfs(0,0))
        {
            puts("YES");
        }
        else
        {
            puts("NO");
        }
        while(!s.empty())
        {
            printf("%d ",s.top());
            s.pop();
        }
        printf("\n");
    }
}