子集和问题

最新推荐文章于 2021-12-12 09:06:15 发布

weixin_30719711

最新推荐文章于 2021-12-12 09:06:15 发布

阅读量265

点赞数

CC 4.0 BY-SA版权

文章标签：数据结构与算法

原文链接：http://www.cnblogs.com/fanhao050109/p/11191823.html

题目描述对于一个给定正整数的集合s={x1,x2,x3…xn}和正整数c，编程计算s的第一个子集s1，使得子集s1的和等于c。 

输入第一行有2个正整数n和c
第二行有n个正整数 
n<7000,c<maxlongint


输出一行数据，按输入的顺序输出，若无解则输出"No Solution!"

样例输入5 10
2 2 6 5 4
样例输出2 2 6
提示这道题需要剪枝才能AC，剪枝方法是计算出后缀和数组sum[i]和最小值数组mina[i].在进行枚举时增加逻辑如下：如果当前已选中元素和加上sum[i]小于c，或者当前已选中元素和加上mina[i]大于c，直接返回即可。 
 
#include<cstdio>
#include<algorithm>
using namespace std;
long long n,m,ans[7001],a[7001],sum[7001],minn[70001];
bool chu=0;
void dfs(int k,int chang,int he)
{
    if(he==m)
    {
        for(int i=0;i<chang;i++)
        {
            printf("%d ",ans[i]);
        }
        chu=1;
        return;
    }
    if((k==n||he+sum[k]<m||he+minn[k]>m))
    return;
     
    if(chu==1)
    return;
    if(he+a[k]<=m)
    {
        ans[chang]=a[k];
        dfs(k+1,chang+1,he+a[k]);
    }
        dfs(k+1,chang,he);
}
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=0;i<n;i++)
    {
        scanf("%d",&a[i]);
    }
    sum[n-1]=a[n-1];minn[n-1]=a[n-1];
    for(int i=n-2;i>=0;i--)
    {
        sum[i]=a[i]+sum[i+1];
        minn[i]=1000000000;
        minn[i]=min(minn[i+1],a[i]);
    }
    dfs(0,0,0);
    if(chu==0)
    {
        printf("No Solution!");
    }
}