nyoj1058 部分和问题

/*
部分和问题
时间限制：1000 ms  |  内存限制：65535 KB
难度：2
描述
给定整数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 <stdio.h>
#include <string.h>
int a[25];
int book[25];
int n,ans;
int sum;
int flag;
void dfs(int i)
{
if(sum>ans||flag)
{
return ;
}
if(sum==ans)
{
flag=1;
printf("YES\n");
for(int j=0;j<n;j++)
{
if(book[j])
{
printf("%d ",a[j]);
}
}
printf("\n");
}
for(int k=i;k<n;k++)
{
sum=sum+a[k];
book[k]=1;
dfs(k+1);
sum=sum-a[k];
book[k]=0;
}
}
int main()
{
while(~scanf("%d%d",&n,&ans))
{
memset(book,0,sizeof(book));
sum=0;
flag=0;
int i;
for(i=0;i<n;i++)
{
scanf("%d",&a[i]);
}
dfs(0);
if(!flag)
{
printf("NO\n");
}
}
return 0;
}