Codechef Xor Queries(可持久化字典树)

题意:
给定一个初始时为空的整数序列(元素由1开始标号)以及一些询问:
类型0:在数组最后加入数字x
类型1:在区间[L,R]中找到数字y,最大化(x^y)
类型2:删除数组最后k个元素
类型3:在区间[L,R]中,统计小于等于x的元素个数
类型4:在区间[L,R]中,找到第k小的数
(M<=5e5,x<=5e5)

---

#include<cstdio>
#include<algorithm>
using namespace std;
const int N=5e5+100;

struct Trie{
    int ch[2],sum,dep;
}T[N*30];
int root[N],sz;

void insert(int &i,int d,int x,int v){
    T[++sz]=T[i],i=sz;
    T[i].dep=d+1;
    T[i].sum+=v;    //此时判断的是上一位的d
    if(d<0) return ;
    int p=(x>>d)&1;
    insert(T[i].ch[p],d-1,x,v);
}

void Query_max_xor(int L,int R,int d,int x,int &ans){
    if(d<0)     return ;
    int p=(x>>d)&1;
    if(T[ T[R].ch[p^1] ].sum-T[ T[L].ch[p^1] ].sum)
        ans+=(p^1)*(1<<d),Query_max_xor(T[L].ch[p^1],T[R].ch[p^1],d-1,x,ans);
    else    ans+=p*(1<<d),Query_max_xor(T[L].ch[p],T[R].ch[p],d-1,x,ans);
}

void Query_equal_to_x(int L,int R,int d,int x,int &ans){
    if(d<0)     {ans+=T[R].sum-T[L].sum;return ;}
    int p=(x>>d)&1;
    if(p==1)  ans+=T[ T[R].ch[0] ].sum-T[ T[L].ch[0] ].sum;
    Query_equal_to_x(T[L].ch[p],T[R].ch[p],d-1,x-p*(1<<d),ans);
}

void Query_kth(int L,int R,int d,int x,int &ans){
    if(d<0)     return ;
    int k=T[ T[R].ch[0] ].sum-T[ T[L].ch[0] ].sum;
    if(k>=x)    Query_kth(T[L].ch[0],T[R].ch[0],d-1,x,ans);
    else   ans+=(1<<d),Query_kth(T[L].ch[1],T[R].ch[1],d-1,x-k,ans);
}

int main(){
    int m,n=0,L,R,ans,op,x;
    while(scanf("%d",&m)!=EOF){
        sz=0,root[0]=0;
        while(m--){
            scanf("%d",&op);
            if(op==0){
                scanf("%d",&x);
                root[++n]=root[n-1];
                insert(root[n],23,x,1);
            }
            else if(op==1){
                scanf("%d%d%d",&L,&R,&x);
                Query_max_xor(root[L-1],root[R],23,x,ans=0);
                printf("%d\n",ans);
            }
            else if(op==2)
                scanf("%d",&x),n-=x;
            else if(op==3){
                scanf("%d%d%d",&L,&R,&x);
                Query_equal_to_x(root[L-1],root[R],23,x,ans=0);
                printf("%d\n",ans);
            }
            else if(op==4){
                scanf("%d%d%d",&L,&R,&x);
                Query_kth(root[L-1],root[R],23,x,ans=0);
                printf("%d\n",ans);
            }
        }
    }
    return 0;
}