部分和问题

部分和问题
时间限制：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<iostream>
#include<cstring>
#include <cstdio>
using namespace std;
int a[25],f,m[25];
int n,k,sum;
void dfs(int cnt)
{
    if (f==0)
    return ;
    if (cnt<=n)
    {
        if (sum==k)
        {
            f=0;
            printf("YES\n");
            for (int i=0;i<=n;++i)
            {
                if (m[i]==1)
                {
                    printf("%d ",a[i]);
                }
            }
            printf("\n");
        }
        for(int i=cnt;i<=n;++i)
        {
            sum+=a[i];
            m[i]=1;
            dfs(i+1);
            sum-=a[i];
            m[i]=0;
        }
    }
}
int main()
{
    int i;
    while(~scanf("%d %d",&n,&k))
    {
        memset(m,0,sizeof(m));
        for (i=0;i<n;++i)
        {
            scanf("%d",&a[i]);
        }
        f=1;sum=0;
        dfs(0);
        if (f==1)
        {
            printf("NO\n");
        }
    }
    return 0;
}

标程：

#include<stdio.h>
int f = 1,n,k,w[21]={0},s[21]={0};
bool a[21]={0};
void DFS(int i, int c)
{
    int j;
    if(i>=0)
    {
        if(c+s[i]<k)return;
        if(c+w[i]<=k)
        {
            a[i]=1;
            DFS(i-1,c+w[i]);
            a[i]=0;         
        }
        DFS(i-1,c);     
        if(c==k&&f)
        {
            f=0;
            printf("YES\n");
            for(j=1;j<=n;j++)
            if(a[j])
            printf("%d ",w[j]);
            printf("\n");
        }
    }
}

int main()
{
    while(~scanf("%d%d",&n,&k))
    {
        int i;f=1;
        for(i=1;i<=n;i++)
        {
            s[i]=0;
            scanf("%d",&w[i]);
            s[i]=s[i-1]+w[i];
        }
        DFS(n,0);
        if(f)printf("NO\n");
    }
}