部分和问题

描述
给定整数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<iostream>
using namespace std;

int a[22] = {0};
int b[22] = {0};
int k;
int n ;

// 已经从前i项得到了和sum 然后对于i项之后的进行分支
bool dfs(int i,int sum) {
   //如果前i项都应经计算过了 则返回与sum是否相等
   if(i==n) return sum==k;
   //不加上a[i]的情况
   if(dfs(i+1,sum)) return true;
   if(dfs(i+1,sum+a[i])) return true;
   else return false;
}
int main() {
    cout << "输入n和k" << endl;
    cin >> n >> k;

    for(int i=0;i<n;i++) {
        cin >> a[i] ;
    }
    if(dfs(0,0))    cout << "YES" << endl;
    else  cout << "NO" << endl;


    return 0;
}