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