AtCoder Regular Contest 100 E - Or Plus Max(高维前缀和sosdp)

题目

给定n(n<=18),长度为2^n的数组a[],a_{0},a_{1},\dots,a_{2^{n}-1},1<=ai<=1e9

对于每个k\epsilon [1,2^{n}-1], 求满足0\leq i<j\leq 2^{n}-1,(i|j)\leq k的ai+aj的最大值

思路来源

https://www.cnblogs.com/heyuhhh/p/11585358.html

题解

(i or j)<=k,很不好做,

考虑转化成(i or j)==k再做前缀和,发现也很难做

于是考虑(i or j)是k的子集,对k统计k的子集中的top2,

再做一遍前缀和,这样,

统计用k的子集更新k的答案,再用k-1的子集,...,1的子集更新k的答案

于是就变成了先高维前缀和保存top2,然后再前缀和更新一遍答案

代码

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=19,M=1<<N;
#define fi first
#define se second
typedef pair<int,int> P;
int n,v,ans[M];
P dp[M];
void merge(P &a,int x){
    if(x>=a.fi){
        a.se=a.fi;
        a.fi=x;
    }
    else if(x>a.se){
        a.se=x;
    }
}
int main(){
    scanf("%d",&n);
    dp[0]=P(-1,-1);
    for(int i=0;i<(1<<n);++i){
        scanf("%d",&v);
        dp[i]=P(v,-1);
    }
    for(int i=0;i<n;++i){
        for(int j=0;j<(1<<n);++j){
            if(j>>i&1){//去掉1位的子集
                merge(dp[j],dp[j^(1<<i)].fi);
                merge(dp[j],dp[j^(1<<i)].se);
            }
        }
    }
    for(int j=1;j<(1<<n);++j){
        ans[j]=dp[j].fi+dp[j].se;
        ans[j]=max(ans[j],ans[j-1]);
        printf("%d\n",ans[j]);
    }
    return 0;
}

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章