[NYOJ1282]部分和问题（入门题）

部分和问题(入门题)

描述
给你n个数(a1,a2,a3…….an) ,是否存在某一些数字加起来等于k,有就输出 “YES”,否则输出 “NO”。
数据范围：n<20;
a1+a2+….an在int范围里面.

输入
多组输入
每组第一行输入两个数n,k
第二行输入n个数a1 a2 …… an

输出
如果存在一些数加起来为k输出”YES”;否则输出”NO”.

样例输入
5 6
2 3 5 2 1
3 6
2 3 9

样例输出
YES
NO
很水，对于当时刚学深搜的我除外

#include<cstdio>
#include<iostream>
#include<cmath>
#include<cstring>
using namespace std;
int n,k,q=0;
int a[25],b[25],f[25];
void print(int m)
{
    cout<<"YES"<<endl;
    for(int i=1;i<=m;i++)
        cout<<b[i]<<" ";
    cout<<endl;
}
void dfs(int t,int sum,int pre)
{
    if(q==1)return;
    for(int i=pre+1;i<=n;i++)
    {
        if(!f[i])
        {
            sum+=a[i];
            b[t]=a[i];
            f[i]=1;
            if(sum==k)
            {   
                q=1;
                print(t);
                return;
            }
            else
                dfs(t+1,sum,i);
            sum-=a[i];
            f[i]=0;
        }
        if(q==1)return;
    }
}
int main()
{
    while(~scanf("%d%d",&n,&k))
    {
        memset(a,0,sizeof(a));
        memset(b,0,sizeof(b));
        memset(f,0,sizeof(f));
        for(int i=1; i<=n; i++)
        scanf("%d",&a[i]);
        dfs(1,0,0);
        if(q==0) cout<<"NO"<<endl;
        q=0;
    }
}