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

来源

经典题目

上传者

TC_杨闯亮

解题：一顿乱搜。。。。。。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <vector>
#include <climits>
#include <algorithm>
#include <cmath>
#define LL long long
using namespace std;
int n,d[21],ans[21],m,res;
bool dfs(int cur,int add,int tot) {
    if(cur >= n) {
        if(add == res) {
            m = tot;
            return true;
        }
        return false;
    }
    if(dfs(cur+1,add,tot)) {
        return true;
    }
    if(dfs(cur+1,add+d[cur],tot+1)) {
        ans[tot] = d[cur];
        return true;
    }
    return false;
}
int main() {
    while(~scanf("%d %d",&n,&res)) {
        int i;
        for(int i = 0; i < n; i++)
            scanf("%d",d+i);
        if(dfs(0,0,0)) {
            puts("YES");
            for(i = 0; i+1 < m; i++)
                printf("%d ",ans[i]);
            printf("%d
",ans[i]);
        } else puts("NO");
    }
    return 0;
}