BZOJ 3689 异或之 (可持久化01Trie+堆)

题目大意：给你一个序列，求出第$K$小的两两异或值

先建出来可持久化$01Trie$

用一个$set$/堆存结构体，存某个异或对$<i,j>$的第二关键字$j$，以及$ai;xor;aj$的值，堆中按异或值从小到大排序

每次取出一对$<i,j>$并把它从堆中删除

在$[0,j-1]$的 可持久化$01Trie$ 中把$a_{i}$这个数删除

再查询$[0,j-1]$中和$a_{j}$的异或最大值，重新推入堆中...

反复操作$K$次即可

删除操作中的细节比较多

#include <set>
#include <queue>
#include <vector>
#include <cstdio>
#include <cstring>
#include <algorithm>
#define N1 351000
#define N2 13000000
#define MM 100
#define ll long long
#define dd double  
#define uint unsigned int
#define mod 1000000007
#define idx(X) (X-'a')
#define it multiset<node>::iterator
using namespace std;

uint bin[35];
struct Trie{
int ch[N2][2],num[N2],root[N1],tot;
void init()
{
    root[0]=tot=1;int x=1;
    for(int i=31;i>=0;i--){
        ch[x][0]=++tot;
        x=ch[x][0],num[x]=1;
    }
}
void insert(uint s,int rt1,int rt2,int w)
{
    int x,y,p;
    y=root[rt1];
    x=root[rt2]=++tot;
    for(int i=31;i>=0;i--)
    {
        p=(s&bin[i])?1:0;
        ch[x][p]=++tot;
        ch[x][p^1]=ch[y][p^1];
        num[ch[x][p]]=num[ch[y][p]]+w;
        x=ch[x][p],y=ch[y][p];
    }
}
uint query(uint s,int l,int r)
{
    int x,y,p;uint ans=0;
    x=root[r],y=l<0?0:root[l];
    for(int i=31;i>=0;i--)
    {
        p=(s&bin[i])?1:0;
        if(num[ch[x][p]]-num[ch[y][p]]>0){
            x=ch[x][p],y=ch[y][p];
        }else if(num[ch[x][p^1]]-num[ch[y][p^1]]>0){
            x=ch[x][p^1],y=ch[y][p^1];
            ans|=bin[i];
        }else break;
    }return ans;
}
}T;

int n,K;
int a[N1],res[N1];
struct node{
    int id;uint val;
    friend bool operator < (const node &s1,const node &s2){
        if(s1.val!=s2.val)
            return s1.val<s2.val;
        else 
            return s1.id<s2.id;
    }
    node(int id,uint val):id(id),val(val){}
    node(){}
};
multiset<node>s;

int main()
{
    //freopen("t1.in","r",stdin);
    scanf("%d%d",&n,&K);
    for(int i=0;i<=31;i++)
        bin[i]=(1<<i);
    T.init();uint x;
    for(int i=1;i<=n;i++){
        scanf("%d",&a[i]);
        T.insert(a[i],i-1,i,1);
    }
    for(int i=2;i<=n;i++){
        x=T.query(a[i],0,i-1);
        s.insert(node(i,x));
        res[i]=i-1;
    }
    node p;int i,w;
    for(int k=1;k<=K;k++){
        it t=s.begin();
        w=(*t).val,i=(*t).id;
        printf("%u
",w);
        T.insert(w^a[i],i-1,i-1,-1);
        w=T.query(a[i],0,i-1);
        s.erase(t++);res[i]--;
        if(res[i]) s.insert(node(i,w));
    }
    return 0;
}