nyoj 部分和问题 深搜dfs

题目链接：http://acm.nyist.edu.cn/JudgeOnline/problem.php?pid=1058

注意剪枝。

AC代码：

#include<cstdio>
#include<iostream>
using namespace std;
int visit[22],p[22];
int flag,sum,n,k;
void dfs(int pos)
{
//    cout<<"pos: "<<pos<<"   visit: ";
//    for(int i=0;i<n;i++)
//    {
//        cout<<visit[i]<<" ";
//    }
//    cout<<"  sum:"<<sum<<endl;
    if(sum==k)
    {
        flag=1;
        printf("YES\n");
        for(int i=0; i<n; i++)
        {
            if(visit[i])
            {
                printf("%d ",p[i]);
            }
        }
        printf("\n");
    }
    else
    {
        for(int i=pos; i<n; i++)
        {
            if(visit[i]==0&&sum<=k)
            {
                visit[i]=1;
                sum+=p[i];
                dfs(i+1);
                visit[i]=0;
                sum-=p[i];
            }
        }
    }
}
int main()
{
    while(scanf("%d%d",&n,&k)!=EOF)
    {
        flag=0,sum=0;
        for(int i=0; i<22; i++)
        {
            visit[i]=0;
        }
        for(int i=0; i<n; i++)
        {
            scanf("%d",&p[i]);
        }
        dfs(0);
        if(flag==0)
        {
            printf("NO\n");
        }
    }
    return 0;
}