[可持久化字典树] BZOJ 4546 codechef XRQRS

这就是个大模板题

本来后两个操作想再打个主席树

后来发现可持久化字典树就是主席树

#include<cstdio>
#include<cstdlib>
#include<algorithm>
using namespace std;
 
inline char nc()
{
    static char buf[100000],*p1=buf,*p2=buf;
    if (p1==p2) { p2=(p1=buf)+fread(buf,1,100000,stdin); if (p1==p2) return EOF; }
    return *p1++;
}
 
inline void read(int &x)
{
    char c=nc(),b=1;
    for (;!(c>='0' && c<='9');c=nc()) if (c=='-') b=-1;
    for (x=0;c>='0' && c<='9';x=x*10+c-'0',c=nc()); x*=b;
}
 
const int N=500005;
const int K=20;
const int M=10921000;
 
int ncnt;
int root[N];
int ch[M][2],sum[M];
 
inline void insert(int &rt,int x,int t){
    rt=++ncnt; 
    int y; sum[y=rt]=sum[x]+1;
    for (int i=K;~i;i--)
    {
        if ((t>>i)&1)
            ch[y][0]=ch[x][0],y=(ch[y][1]=++ncnt),x=ch[x][1];
        else
            ch[y][1]=ch[x][1],y=(ch[y][0]=++ncnt),x=ch[x][0];
        sum[y]=sum[x]+1;
    }
}
 
inline int Max(int l,int r,int t){
    int ret=0,tmp;
    for (int i=K;~i;i--)
    {
        tmp=(t>>i)&1;
        if (sum[ch[r][tmp^1]]-sum[ch[l][tmp^1]])
            ret+=((tmp^1)<<i),r=ch[r][tmp^1],l=ch[l][tmp^1];
        else
            ret+=(tmp<<i),r=ch[r][tmp],l=ch[l][tmp];
    }
    return ret;
}
 
inline int Count(int l,int r,int t){
    int ret=0,tmp;
    for (int i=K;~i;i--)
    {
        tmp=(t>>i)&1;
        if (tmp)
            ret+=sum[ch[r][0]]-sum[ch[l][0]],l=ch[l][1],r=ch[r][1];
        else
            l=ch[l][0],r=ch[r][0];
    }
    ret+=sum[r]-sum[l];
    return ret;
}
 
inline int Kth(int l,int r,int k){
    int ret=0;
    for (int i=K;~i;i--)
        if (k>sum[ch[r][0]]-sum[ch[l][0]])
            k-=sum[ch[r][0]]-sum[ch[l][0]],ret+=(1<<i),l=ch[l][1],r=ch[r][1];
        else
            l=ch[l][0],r=ch[r][0];
    return ret;
}
 
int pnt;
 
int main()
{
    int Q,order,l,r,x;
    read(Q);
    while (Q--)
    {
        read(order); 
        if (order==1)
        {
            read(x); pnt++;
            insert(root[pnt],root[pnt-1],x);
        }
        else if (order==2)
        {
            read(l); read(r); read(x);
            printf("%d\n",Max(root[l-1],root[r],x));
        }
        else if (order==3)
        {
            read(x);
            pnt-=x;
        }
        else if (order==4)
        {
            read(l); read(r); read(x);
            printf("%d\n",Count(root[l-1],root[r],x));
        }
        else if (order==5)
        {
            read(l); read(r); read(x);
            printf("%d\n",Kth(root[l-1],root[r],x));
        }
    }
    return 0;
}