NYOJ 1058 部分和问题


部分和问题

时间限制: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
来源
经典题目
上传者
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");
    }
}



展开阅读全文

没有更多推荐了,返回首页