Count on a tree

树上的主席树。
从父节点建树。
容斥原理 root[a]+root[b]-root[lca(a,b)]-root[fa[lca(a,b)]] 上的第k大。

#include <cstdio>
#include <iostream>
#include <algorithm>
#define il inline
#define ls l,mid
#define rs mid+1,r
using namespace std;
const int maxm=410000;
int val[maxm],hash[maxm];
int head[maxm],net[2*maxm],to[2*maxm];
int cnt,t;
int dep[maxm],fa[maxm][17];
int pos[maxm],id[maxm],ind;
int root[maxm];
int sum[maxm<<5],lson[maxm<<5],rson[maxm<<5],size;
int build(int l,int r)
{
    int rt=++size;
    sum[rt]=0;
    if(l<r)
    {
        int mid=(l+r)>>1;
        rson[rt]=build(rs);
        lson[rt]=build(ls);
    }
    return rt;
}
void add(int x,int y)
{
    cnt++;
    to[cnt]=y;
    net[cnt]=head[x];
    head[x]=cnt;
} 
il int read()
{
    int x=0,w=1;
    char ch=0;
    while(ch<'0'||ch>'9')
    {
        if(ch=='-') w=-1;
        ch=getchar();
    }
    while(ch>='0'&&ch<='9')
    {
        x=(x<<3)+(x<<1)+ch-'0';
        ch=getchar();
    }
    return x*w;
}
int updata(int pre,int l,int r,int v)
{
    int rt=++size;
    lson[rt]=lson[pre],rson[rt]=rson[pre],sum[rt]=sum[pre]+1;
    if(l<r)
    {
        int mid=(l+r)>>1;
        if(v<=mid) lson[rt]=updata(lson[pre],ls,v);
        else rson[rt]=updata(rson[pre],rs,v);
    }
    return rt;
}
void dfs(int x)
{
    ind++,id[x]=ind,pos[ind]=x;
    for(int i=1;i<=16&&(1<<i)<=dep[x];i++)
     fa[x][i]=fa[fa[x][i-1]][i-1];
    root[id[x]]=updata(root[fa[x][0]],1,t,val[x]);
    for(int i=head[x];i;i=net[i])
     if(to[i]!=fa[x][0]) fa[to[i]][0]=x,dep[to[i]]=dep[x]+1,dfs(to[i]);
}
il int LCA(int x,int y)
{
    if(dep[x]<dep[y]) swap(x,y);
    int t=dep[x]-dep[y];
    for(int i=0;i<=16;i++)
     if((1<<i)&t) x=fa[x][i];
    for(int i=16;i>=0;i--)
     if(fa[x][i]!=fa[y][i])
      x=fa[x][i],y=fa[y][i];
    if(x==y) return x;
    return fa[x][0];
}
il int ask(int a,int b,int c,int d,int l,int r,int kth)
{
   if(l>=r) return hash[l];
   int nowsum=sum[lson[a]]+sum[lson[b]]-sum[lson[c]]-sum[lson[d]];
   int mid=(l+r)>>1;
   if(nowsum>=kth) return ask(lson[a],lson[b],lson[c],lson[d],ls,kth);
   else return ask(rson[a],rson[b],rson[c],rson[d],rs,kth-nowsum);
}
int main()
{
    int n,m;
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)
     scanf("%d",&val[i]),hash[i]=val[i];
    sort(hash+1,hash+n+1);
    t=unique(hash+1,hash+n+1)-hash-1;
    for(int i=1;i<=n;i++)
     val[i]=lower_bound(hash+1,hash+t+1,val[i])-hash;
    for(int i=1,u,v;i<n;i++)
     scanf("%d%d",&u,&v),add(u,v),add(v,u);
    dfs(1); 
    //root[0]=build(1,t);
    int last=0;
    for(int i=1;i<=m;i++)
    {
        int u,v,k;
        scanf("%d%d%d",&u,&v,&k);
        u^=last;
        int lca=LCA(u,v);
        last=ask(root[id[u]],root[id[v]],root[id[lca]],root[id[fa[lca][0]]],1,t,k);
        printf("%d",last);
        if(i!=m) puts("");
    }

    return 0;
}