nyoj 1058 部分和问题

描述

给定整数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>
#include<cstdio>
#include<cstring>
using namespace std;
    int n,k,f,sum;
    int a[25],b[25];
void dfs(int x){
   int i;
   if(sum>=k){
    if(sum==k){
     if(f){
            f=0;
            printf("YES\n");
     }
      for(i=0;i<n;i++){
        if(b[i])
            printf("%d ",a[i]);
      }
      printf("\n");
   }
    return ;
   }
    for(i=x;i<n;i++){
        sum+=a[i];
        b[i]=1;
        dfs(i+1);
        b[i]=0;
        sum-=a[i];
    }
}
int main(){
    int i;
    while(scanf("%d%d",&n,&k)!=EOF){
        for(i=0;i<n;i++){
            scanf("%d",&a[i]);
        }
        f=1;
        sum=0;
        memset(b,0,sizeof(b));
        dfs(0);
        if(f)
            printf("NO\n");

    }
    return 0;
}