NYOJ-1058 DFS部分求和

 原题链接

本题思路：暴力深搜，具体见代码。

#include <iostream>
#include <stack>
using namespace std;

const int maxn = 20 + 5;
int a[maxn];
int n, k;
bool flag;
stack <int > s;

bool dfs(int i, int sum) {
    if(i == n) return sum == k;//当搜到最后一个数时返回sum == k
    if(dfs(i + 1, sum))  return true;//不加这个数的情况
    if(dfs(i + 1, sum + a[i])) {//如果添加这个数满足k == sum 则将这个数加入栈中
        s.push(a[i]);
        return true;
    }
    return false;
}

int main () {
    while(cin >> n >> k) {
        flag = true;
        if(n == 0)  break;
        for(int i = 0; i < n; i ++)
            cin >> a[i];
        flag = dfs(0, 0);
        if(flag) {
            cout << "YES" << endl;
            while(!s.empty()) {
                cout << s.top() << " ";
                s.pop();
            }
        } else cout << "NO" << endl;
    }
    return 0;
}