NYOJ 1058 部分和问题

部分和问题

时间限制：1000 ms  |  内存限制：65535 KB

难度：2

描述

给定整数a1、a2、.......an，判断是否可以从中选出若干数，使它们的和恰好为K。

输入

首先，n和k，n表示数的个数，k表示数的和。
接着一行n个数。
（1<=n<=20,保证不超int范围）

输出

如果和恰好可以为k，输出“YES”，并按输入顺序依次输出是由哪几个数的和组成，否则“NO”

样例输入

4 131 2 4 7

样例输出

YES2 4 7

来源

经典题目

上传者

TC_杨闯亮

经历：首先这题过得很憋屈，这题据说要多组输入，但我刚开始就多组输入了，所以这一点没有证实。还有就是当有多组数据满足条件时，只需要输出第一组满足条件的解即可！！！（就是这里卡了好久，把所有满足的解都输出来了）

#include<stdio.h>#include<string.h>int n,k,flag,a[29],vis[30],ans[30],sum;void dfs(int x,int y){    if(sum>k||y==n)        return;    if(sum==k)    {        flag=0;        printf("YES\n");        for(int i=0;i<=x;i++)            printf(i==x?"%d\n":"%d ",ans[i]);        return;    }    for(int i=y+1;i<n;i++)    {        if(vis[i]==0)        {            sum+=a[i];            ans[x+1]=a[i];            vis[i]=1;            dfs(x+1,y+1);            if(!flag)                break;            sum-=a[i];            vis[i]=0;        }    }    return;}int main(){    while(~scanf("%d%d",&n,&k))    {        flag=1;        for(int i=0;i<n;i++)            scanf("%d",&a[i]);        memset(vis,0,sizeof(vis));        for(int i=0;i<n;i++)        {            sum=0;            if(vis[i]==0)            {                sum+=a[i];                ans[0]=a[i];                vis[i]=1;                dfs(0,i);                if(!flag)                    break;                sum-=a[i];            }        }        if(flag)            printf("NO\n");    }}