[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;
    }
}