BZOJ2588 主席树

几乎没怎么写过主席树 今天看着hzwer的写了写 各种RE
$对于询问(u,v,k)$
$a=u,b=v,c=LCA(u,v),d=father(lca)$
在主席树上二分$s=sum[ls[a]]+sum[ls[b]]-sum[ls[c]]-sum[ls[d]];$
$s\leq k \to a,b,c,d=ls(a),ls(b),ls(c),ls(d)$
$s>k \to a,b,c,d=rs(a),rs(b),rs(c),rs(d)$

#include<bits/stdc++.h>
#define hash _hash
using namespace std;
const int N=2e5+5;
const int M=4e6+5;
inline int read(){
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
struct Edge{
    int v,nxt;
}e[N*2];
int ls[M],rs[M],sum[M],dep[N],f[N][18],head[N],bin[18]={1};
int n,m,tot,cnt,sz,lastans;
int v[N],tmp[N],hash[N],root[N];
inline void add(int u,int v){
    e[++cnt].v=v,e[cnt].nxt=head[u],head[u]=cnt;
    e[++cnt].v=u,e[cnt].nxt=head[v],head[v]=cnt;
}
inline void update(int l,int r,int x,int &y,int num){
    y=++sz,sum[y]=sum[x]+1;
    if(l==r) return;
    ls[y]=ls[x],rs[y]=rs[x];
    int mid=(l+r)>>1;
    if(num<=mid) update(l,mid,ls[x],ls[y],num);
    else update(mid+1,r,rs[x],rs[y],num);
} 
inline void dfs(int x,int ff){
    update(1,tot,root[ff],root[x],v[x]);
    f[x][0]=ff,dep[x]=dep[ff]+1;
    for(int i=head[x];i;i=e[i].nxt){
        int j=e[i].v;
        if(j==ff) continue;
        dfs(j,x);
    }
}
inline void findf(){
    for(int i=1;i<=16;i++) for(int j=1;j<=n;j++) if(dep[j]>=bin[i])f[j][i]=f[f[j][i-1]][i-1];
}
inline int LCA(int x,int y){
    if(dep[x]<dep[y]) swap(x,y);
    int set=dep[x]-dep[y];
    for(int i=16;i>=0;i--) if(set>=bin[i]) {
        x=f[x][i];
        set-=bin[i];
    }
    if(x==y) return x;
    for(int i=16;i>=0;i--) if(f[x][i]!=f[y][i]){
        x=f[x][i],y=f[y][i];
    }
    return f[x][0];
}
inline int find(int x){
    int l=1,r=tot;
    while(l<=r){
        int mid=(l+r)>>1;
        if(hash[mid]<x)l=mid+1;
        else if(hash[mid]==x)return mid;
        else r=mid-1;
    }
}
inline int query(int a,int b,int c,int d,int k){
    int l=1,r=tot; 
    while(l<r){
        int mid=(l+r)>>1;
        int zjq=sum[ls[a]]+sum[ls[b]]-sum[ls[c]]-sum[ls[d]];
        if(zjq>=k) r=mid,a=ls[a],b=ls[b],c=ls[c],d=ls[d];
        else k-=zjq,l=mid+1,a=rs[a],b=rs[b],c=rs[c],d=rs[d];
    }
    return l;
}
int main(){
#ifdef Devil_Gary
    freopen("in.txt","r",stdin);
#endif
    for(int i=1;i<=16;i++) bin[i]=bin[i-1]*2;
    n=read(),m=read();
    for(int i=1;i<=n;i++) v[i]=read(),tmp[i]=v[i];
    sort(tmp+1,tmp+n+1);
    hash[++tot]=tmp[1];
    for(int i=2;i<=n;i++)if(tmp[i]!=tmp[i-1])hash[++tot]=tmp[i];
    for(int i=1;i<=n;i++)v[i]=find(v[i]);
    for(int i=1;i<n;i++) add(read(),read());
    dfs(1,0);
    findf();
    for(int i=1,x,y,z;i<=m;i++){
        x=read()^lastans,y=read(),z=read();
        int lca=LCA(x,y),flca=f[lca][0];
        printf("%d",lastans=hash[query(root[x],root[y],root[lca],root[flca],z)]);
        if(i!=m)printf("\n");
    }
    return 0;
}