nyoj 1058部分和问题（DFS）

部分和问题

时间限制：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<cstdio>
#include<cstring>
bool vis[22], ok;
int a[22];
int n, k;
void dfs(int sum, int cur)
{
if(sum==k)
{
if(!ok)
{
ok = 1;
printf("YES\n");
}
for(int i=0; i<n; i++)
if(vis[i])
printf("%d ", a[i]);
printf("\n");
return;
}
for(int i=cur; i<n; i++)
{
sum+=a[i];
vis[i] = 1;
dfs(sum, i+1);
vis[i] = 0;
sum-=a[i];
}
}

int main()
{
while(scanf("%d%d", &n, &k)!=EOF)
{
for(int i=0; i<n; i++)
scanf("%d", &a[i]);
memset(vis, 0, sizeof(vis));
ok = 0;
dfs(0, 0);
if(ok==0)
printf("No\n");
}
return 0;
}

此题有更优的解法-------01背包。