折半枚举小结-CSDN博客

本文链接：https://blog.csdn.net/qq_43555854/article/details/108875707

题目链接
首先明确一件事，折半枚举不是二分！不是二分！不是二分！虽然我第一次听到这个名词也以为是二分23333
2019牛客多校第九场的签到题。。
看到群里有人问道题，别人跟他说折半枚举。。做题做到现在从来没听说过折半枚举，真是惭愧，于是赶紧去学了一下。
入门题是这一道，给你很多个n个pair<a,b,c,d>,问你现在能选多少组相同或不同pair里的a，b，c，d使他们的和为0,n<4000。
首先暴力的n^4做法是不可取的，所以要给他优化，折半枚举就是先把每对c+d的结果先给预处理出来，然后只要n^2枚举a+b,然后logn找目标c+d的数量就行(小技巧：用upper_bound-lower_bound算数量)

const int N=4100;
int n,a[N],b[N],c[N],d[N],res[N*N],tot;
int main(){
    tot=0;
    cin>>n;
    for(int i=0; i<n; ++i) cin>>a[i]>>b[i]>>c[i]>>d[i];
    for(int i=0; i<n; ++i){
        for(int j=0; j<n; ++j){
            res[tot++]=c[i]+d[j];
        }
    }
    sort(res,res+tot);
    long long ans=0;
    for(int i=0; i<n; ++i){
        for(int j=0; j<n; ++j){
            int x=-a[i]-b[j];
            ans+=upper_bound(res,res+tot,x)-lower_bound(res,res+tot,x);
        }
    }
    cout<<ans<<'\n';
    return 0;
}

多校的那道题处理的是任意取多少元素满足需求，也是比较常见的类型（主要是因为入门的那种类型太简单了…），已知我们可以通过状态压缩把枚举所有情况的复杂度压缩到2^n * n。
这道题问任取随便多少个元素，使这些元素的和为s，题目给的元素数量为(<=36),那么很明显直接状态压缩枚举是会TLE的，2^20大概是1e6，2^30次方大概是1e9，这里折半枚举就很好地满足了我们的需求，先把前n/2(最多18),2^n * n时间内算出前半段所有的取值可能，然后sort一下，方便之后的二分查找。然后就可以枚举后半段了，对于每种情况二分找前半段有没有加起来符合和为s的情况，有的话就输出并break（可break可不break，因为题目保证答案唯一）。

const ll N=(ll)1e7;
ll n,s,a[N],tot=0;
struct node{
    ll val,str;
    node(){}
    node(ll _val,ll _str):val(_val),str(_str){}
}res[N];
bool cmp(node n1, node n2){
    return n1.val<n2.val;
}
int solve(ll x){
    ll l=0,r=tot-1,ans=-1;
    while(l<=r){
        int mid=l+(r-l)/2;
        if(res[mid].val==x) return res[mid].str;
        else if(res[mid].val<x) l=mid+1;
        else{
            r=mid-1;
        }
    }
    if(ans==-1) return -1;
    else return res[ans].str;
}
void preout(ll x){
    for(int i=0; i<n/2; ++i){
        if(x&(1<<i)) cout<<1;
        else cout<<0;
    }
}
void aftout(ll x){
    for(int i=n/2; i<n; ++i){
        int act=i-n/2;
        if(x&(1<<act)) cout<<1;
        else cout<<0;
    }
}
int main() {
    cin>>n>>s;
    forn(i,n) cin>>a[i];
    for(int i=0; i<(1<<(n/2)); ++i){
        ll sum=0;
        for(int j=0; j<n/2; ++j){
            if(i&(1<<j)){
                sum+=a[j];
            }
        }
        res[tot++]=node(sum,i);
    }
    sort(res,res+tot,cmp);
    for(int i=0; i<(1<<(n-n/2)); ++i){
        ll sum=0;
        for(int j=0; j<(n-n/2); ++j){
            if(i&(1<<j)) sum+=a[j+n/2];
        }
        ll goal=s-sum;
        ll ret=solve(goal);
        if(ret!=-1){
            preout(ret); aftout(i);
            cout<<'\n';
            break;
        }
    }
}