【bzoj 2588】Count on a tree（可持久化线段树+LCA）

传送门biu~
树上每个节点建一棵权值线段树记录这个点到根的所有点的权值，根据父节点可持久化。
查询(a,b)时，查询的是“线段树a+线段树b-线段树lca(a,b)-线段树father[lca(a,b)]”，在权值线段树上求第k小即可。

#include<bits/stdc++.h>
using namespace std;
const int N=100005,inf=~0u>>1;
int n,m,ans,a[N],dep[N],fa[N][18];
int head[N],nex[2*N],to[2*N],tp;
struct Node{
    Node *ch[2];
    int sum;
    Node();
    inline void maintain(){
        sum=0;
        if(ch[0])   sum+=ch[0]->sum;
        if(ch[1])   sum+=ch[1]->sum;
    }
}*null=new Node,*root[N];
Node :: Node(){
    ch[0]=ch[1]=null;
    sum=0;
}
void buildtree(Node *&o,Node *u,int l,int r,int x){
    o=new Node;
    if(l==r){
        o->sum=u->sum+1;
        return;
    }
    int mid=1ll*l+r>>1;
    if(x<=mid)  buildtree(o->ch[0],u->ch[0],l,mid,x),o->ch[1]=u->ch[1];
    else        buildtree(o->ch[1],u->ch[1],mid+1,r,x),o->ch[0]=u->ch[0];
    o->maintain();
}
int Query(Node *a,Node *b,Node *c,Node *d,int l,int r,int k){
    if(l==r)    return r;
    int mid=1ll*l+r>>1;
    int sum=a->ch[0]->sum+b->ch[0]->sum-c->ch[0]->sum-d->ch[0]->sum;
    if(sum>=k)  return Query(a->ch[0],b->ch[0],c->ch[0],d->ch[0],l,mid,k);
    else return Query(a->ch[1],b->ch[1],c->ch[1],d->ch[1],mid+1,r,k-sum);
}
inline void add(int x,int y){
    nex[++tp]=head[x];
    head[x]=tp;
    to[tp]=y;
}
void dfs(int x){
    dep[x]=dep[fa[x][0]]+1;
    buildtree(root[x],root[fa[x][0]],-inf,inf,a[x]);
    for(int i=head[x];i;i=nex[i]){
        if(to[i]==fa[x][0]) continue;
        fa[to[i]][0]=x;
        dfs(to[i]);
    }
}
inline int lca(int x,int y){
    if(dep[x]<dep[y])   swap(x,y);
    for(int i=17;i>=0;--i)
        if(dep[fa[x][i]]>=dep[y]) x=fa[x][i];
    if(x==y)    return x;
    for(int i=17;i>=0;--i){
        int xn=fa[x][i],yn=fa[y][i];
        if(xn!=yn)  x=xn,y=yn;
    }
    return fa[x][0];
}
int main(){
    null->ch[0]=null->ch[1]=null;
    root[0]=new Node;
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;++i)   scanf("%d",&a[i]);
    for(int i=1;i<n;++i){
        int u,v;
        scanf("%d%d",&u,&v);
        add(u,v);add(v,u);
    }
    dfs(1);
    for(int j=1;j<=17;++j)
        for(int i=1;i<=n;++i)
            fa[i][j]=fa[fa[i][j-1]][j-1];
    while(m--){
        int a,b,c,d,k;
        scanf("%d%d%d",&a,&b,&k);
        a^=ans;c=lca(a,b);d=fa[c][0];
        ans=Query(root[a],root[b],root[c],root[d],-inf,inf,k);
        printf("%d",ans);
        if(m)puts("");
    }
    return 0;
}