Codeforces 706D Vasiliy's Multiset(可持久化字典树)

【题目链接】  http://codeforces.com/problemset/problem/706/D

 

【题目大意】

    要求实现一个集合中的三个操作,1:在集合中加入一个元素x,2:从集合中删除一个元素x(保证x存在),3:要求从集合中选出一个数,使得其与给出的数x的异或值最大,输出这个异或值。

【题解】

  可以将所有的以二进制形式存在01字典树上,删除即插入权值为-1的二进制串,对于异或值最大的操作,我们只要在字典树上按位贪心,从最高位开始尽量保证该位存在最后就能得到答案。写代码的时候直接写了个持久化的字典树,普通字典树也可以实现这样子的功能。

【代码】

#include <cstdio>
#include <algorithm> 
const int N=600005,inf=2000000000;
using namespace std;
int bin[32],n,m,a[N],b[N],root[N];
struct trie{
    int cnt,ch[N*32][2],sum[N*32];
    int insert(int x,int val,int d){
        int tmp,y;tmp=y=++cnt;
        for(int i=30;i>=0;i--){
        int t=val&bin[i];t>>=i;
            ch[y][0]=ch[x][0];ch[y][1]=ch[x][1];
        x=ch[x][t]; y=ch[y][t]=++cnt;
        sum[y]=sum[x]+d; 
        }return tmp;
    }
    int query(int l,int r,int val){
        int tmp=0;
        for(int i=30;i>=0;i--){
        int t=val&bin[i];t>>=i;
        if(sum[ch[r][t^1]]-sum[ch[l][t^1]])tmp+=bin[i],r=ch[r][t^1],l=ch[l][t^1];
        else r=ch[r][t],l=ch[l][t];
        }return tmp;
    }
}T;
int Q,x;char t;
int main(){ 
    bin[0]=1;for(int i=1;i<=30;i++)bin[i]=bin[i-1]<<1;
    scanf("%d",&Q);
    root[1]=T.insert(root[0],0,1);
    for(int i=2;i<=Q+1;i++){
        scanf(" %c %d",&t,&x);
    if(t=='+')root[i]=T.insert(root[i-1],x,1);
    else if(t=='-')root[i]=T.insert(root[i-1],x,-1);
    else printf("%d
",T.query(root[0],root[i]=root[i-1],x));
    }return 0;
}

  

原文地址:https://www.cnblogs.com/forever97/p/codeforces706d.html