zoukankan      html  css  js  c++  java
  • 折半枚举(超大背包)2019牛客多校第九场 D题 组合总和为sum

    https://ac.nowcoder.com/login?callBack=%2Facm%2Fcontest%2F889%2FD%3F%26headNav%3Dacm

    题意:给定长度为n 的数组{a i }和总和sum。

    请找到{a i }的子集,使得子集的总和为sum,输出选取方案的01序列。

    分析:可以看成超大背包,当成普通背包做的话时间复杂度是n*a[ i ] (max) ,而 a[i] 太大,显然会超时,当成暴力枚举来做的话,2^n 也超时,但是n只有36,利用折半枚举,在指数上折半的话就可以大幅减少时间复杂度。

    在进行第二次枚举的查找时,因为数太大就用了set,以及set的find() != set.end() 来实现查找是否存在,而在输出选取方案的01序列时,因为还得输出第一次枚举的方案,所以第一次枚举时用map来保存了state和sum。

    #include<cstdio>
    #include<cstring>
    #include<iostream>
    #include<algorithm>
    #include<set>
    #include<map>
    using namespace std;
    
    typedef long long ll;
    int n,q,p;
    ll sum;
    ll a[39];
    set<ll> se;
    map<ll,int> ma;
    
    int main(){
        scanf("%d%lld",&n,&sum);
        p=n>>1,q=n-(n>>1);
        for(int i=1; i<=n; i++){
            scanf("%lld",a+i);
        }
        for(int i=1; i<(1<<p); i++){
            ll res=0;
            for(int j=0; j<p; j++){
                if(i&(1<<j))
                    res+=a[j+1];
            }
            se.insert(res);
            ma[res]=i;
        }
        for(int i=1; i<(1<<q); i++){
            ll res=0;
            for(int j=0; j<q; j++){
                if(i&(1<<j))
                    res+=a[p+j+1];
            }
            if(se.find(sum-res)!=se.end()){
                int state=ma[sum-res];
                for(int k=0; k<p; k++){
                    if(state&(1<<k)) putchar('1');
                    else putchar('0');
                }
                for(int k=0; k<q; k++){
                    if(i&(1<<k)) putchar('1');
                    else putchar('0');
                }
                puts("");
                return 0;
            }
        }
    }
  • 相关阅读:
    扑克牌顺子
    反转字符串
    左旋转字符串
    和为S的两个数
    C++中substr()详解
    STL库中的equal_range()
    和为S的连续正序列
    数组中只出现一次的数
    二叉树的深度
    mysql找安装路经,更改密码
  • 原文地址:https://www.cnblogs.com/-Zzz-/p/11416746.html
Copyright © 2011-2022 走看看