bzoj5338 [TJOI2018]xor（可持久化Trie）

可持久化Trie裸题。
维护到根的一棵解决路径，维护dfs序的一棵解决子树。
考试时内存开小了gg

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
#define ll long long
#define inf 0x3f3f3f3f
#define N 100010
inline char gc(){
    static char buf[1<<16],*S,*T;
    if(S==T){T=(S=buf)+fread(buf,1,1<<16,stdin);if(T==S) return EOF;}
    return *S++;
}
inline int read(){
    int x=0,f=1;char ch=gc();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=gc();}
    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=gc();
    return x*f;
}
int n,m,h[N],num=0,rt[N],rt2[N],a[N],in[N],out[N],dfn=0;
int fa[N][18],dep[N],Log[N],bin[32],owo=0,w[N];
struct node{
    int son[2],sz;
}tr[N*32*2];
struct edge{
    int to,next;
}data[N<<1];
inline void ins(int &root,int val){
    tr[++owo]=tr[root];root=owo;int p=owo;tr[p].sz++;
    for(int i=30;i>=0;--i){
        bool op=val&bin[i];
        tr[++owo]=tr[tr[p].son[op]];tr[p].son[op]=owo;p=owo;tr[p].sz++;
    }
}
inline void dfs(int x){
    for(int i=1;i<=Log[n];++i){
        if(!fa[x][i-1]) break;
        fa[x][i]=fa[fa[x][i-1]][i-1];
    }in[x]=++dfn;w[dfn]=a[x];rt[x]=rt[fa[x][0]];ins(rt[x],a[x]);
    for(int i=h[x];i;i=data[i].next){
        int y=data[i].to;if(y==fa[x][0]) continue;
        fa[y][0]=x;dep[y]=dep[x]+1;dfs(y);
    }out[x]=dfn;
}
inline int ask(int p1,int p2,int val){
    int res=0;
    for(int i=30;i>=0;--i){
        bool op=val&bin[i];op^=1;
        int sz=tr[tr[p1].son[op]].sz-tr[tr[p2].son[op]].sz;
        if(sz>=1) res|=bin[i],p1=tr[p1].son[op],p2=tr[p2].son[op];
        else p1=tr[p1].son[op^1],p2=tr[p2].son[op^1];
    }return res;
}
inline int ask2(int p1,int p2,int p3,int p4,int val){
    int res=0;
    for(int i=30;i>=0;--i){
        bool op=val&bin[i];op^=1;
        int sz=tr[tr[p1].son[op]].sz+tr[tr[p2].son[op]].sz-tr[tr[p3].son[op]].sz-tr[tr[p4].son[op]].sz;
        if(sz>=1) res|=bin[i],p1=tr[p1].son[op],p2=tr[p2].son[op],p3=tr[p3].son[op],p4=tr[p4].son[op];
        else p1=tr[p1].son[op^1],p2=tr[p2].son[op^1],p3=tr[p3].son[op^1],p4=tr[p4].son[op^1];
    }return res;
}
inline int lca(int x,int y){
    if(dep[x]<dep[y]) swap(x,y);
    int d=dep[x]-dep[y];
    for(int i=0;i<=Log[d];++i)
        if(d&bin[i]) x=fa[x][i];
    if(x==y) return x;
    for(int i=Log[n];i>=0;--i)
        if(fa[x][i]!=fa[y][i]) x=fa[x][i],y=fa[y][i];
    return fa[x][0];
}
int main(){
//  freopen("xor.in","r",stdin);
//  freopen("xor.out","w",stdout);
    n=read();m=read();Log[0]=-1;bin[0]=1;
    for(int i=1;i<=30;++i) bin[i]=bin[i-1]<<1;
    for(int i=1;i<=n;++i) a[i]=read(),Log[i]=Log[i>>1]+1;
    for(int i=1;i<n;++i){
        int x=read(),y=read();
        data[++num].to=y;data[num].next=h[x];h[x]=num;
        data[++num].to=x;data[num].next=h[y];h[y]=num;
    }dfs(1);for(int i=1;i<=n;++i) rt2[i]=rt2[i-1],ins(rt2[i],w[i]);
    while(m--){
        int op=read(),x=read(),y=read();
        if(op==1){
            printf("%d\n",ask(rt2[out[x]],rt2[in[x]-1],y));
        }else{
            int t=lca(x,y);
            printf("%d\n",ask2(rt[x],rt[y],rt[t],rt[fa[t][0]],read()));
        }
    }return 0;
}