【BZOJ 3689】异或之 trie+堆
阅读原文时间:2021年04月20日阅读:1

以前做过类似的题目,不同的是把异或改成加法。。。。。。

还以为异或和加法没有太大的区别,把初始的几个数字放进优先队列里面,一个一个的取出来,得到次大值,再放进去,一直重复知道k个,至于找一个数和其他树异或的k大值可以用trie树轻松解决。

原因嘛,很显然对于每一个数经过我们构造出来的数字是单调递增的,然后放进堆里面每次保证取出堆里面的数字也是单调递增的(有点类似于dijkstra的样子,当时是自己yy的啦)

#include<cstdio>
#include<cstring>
#include<iostream>
#include<queue>
#define maxn 3000211
using namespace std;
int n,k,ch[maxn][2],size[maxn],tot,a[maxn];
struct node{
    int v,k,a;
    bool operator<(const node& y)const{return v>y.v;}
};
priority_queue<node>q;
void insert(int x){
    int rt=0;
    for(int c,i=31;i>=0;i--){
        c=(x>>i)&1;
        if(!ch[rt][c])ch[rt][c]=++tot;
        rt=ch[rt][c];size[rt]++;
    }
}
int query(int x,int k){
    int ans=0,rt=0;
    for(int c,i=31;i>=0;i--){
        c=(x>>i)&1;
        if(k>size[ch[rt][c]]){
            k-=size[ch[rt][c]];
            ans=(ans<<1)|1,rt=ch[rt][!c];
        }else{
            ans<<=1,rt=ch[rt][c];
        }
    }
    return ans;
}
int main(){
    scanf("%d%d",&n,&k);
    for(int i=1;i<=n;i++)scanf("%d",a+i),insert(a[i]);
    for(int i=1;i<=n;i++)q.push((node){query(a[i],2),2,a[i]});
    for(int cnt=0,i=1;;i++){
        node u=q.top();q.pop();
        if(i&1)printf("%d ",u.v),cnt++;
        if(cnt==k)break;if(u.k==n)continue;
        q.push((node){query(u.a,u.k+1),u.k+1,u.a}); }
    return 0;
}

手机扫一扫

移动阅读更方便

阿里云服务器
腾讯云服务器
七牛云服务器

你可能感兴趣的文章