TZOJ4959:部分和问题（深搜DFS、细节）

部分和问题

题目传送门

描述

给定整数a1、a2、…an，判断是否可以从中选出若干数（1个或以上），使它们的和恰好为K。

输入

输入数据有多组，每组第一行为两个整数n和k，n表示数的个数，k表示数的和。
第二行为n个数。

1<=n<=20,-10^8<=ai, k<=10^8

输出

如果和恰好可以为k，输出“Yes”，否则“No”

样例输入

4 15
1 2 4 7

样例输出

No

解题思路

很明显，这道题目考察的是深搜，网上的大多数代码都不足以通过OJ上的毒瘤数据，记录一下踩坑。注意至少选择了一个时才可以更新标记。

代码

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define pii pair<ll,ll>
#define inf 0x3f3f3f3f
#define IOS ios::sync_with_stdio(0), cin.tie(0)
const ll N = 1e1 * 2 + 5;
ll n, k, a[N], f;
void dfs(ll x, ll s) {
    // 注意这里x不等于0，也就是至少选择了一个时才可以更新标记
    if (s == k && x) { 
        f = 1;
        return;
    }
    for (ll i = x; i < n; ++i) {
        dfs(i + 1, s + a[i]);
        if (f) return;
    }
}
void solve() {
    while (cin >> n >> k) {
        f = 0;
        for (int i = 0; i < n; ++i) {
            cin >> a[i];
        }
        dfs(0, 0);
        if (f) cout << "Yes" << endl;
        else cout << "No" << endl;
    }
}

signed main( ){
    IOS;
//    ll t; cin >> t;
    ll t = 1;
    while (t--) solve();
    return 0;
}