主席树 || 可持久化线段树 || LCA || BZOJ 2588: Spoj 10628. Count on a tree || Luogu P2633 Count on a tree

题面： Count on a tree

题解：

主席树维护每个节点到根节点的权值出现次数，大体和主席树典型做法差不多，对于询问（X，Y），答案要计算ans(X)+ans(Y)-ans（LCA(X,Y))-ans(father[LCA(X,Y)])

代码：

 #include<cstdio>
 #include<cstring>
 #include<iostream>
 #include<algorithm>
 using namespace std;
 const int maxn=+,maxm=maxn,maxlog=;
 int lastans=,num_edge=,edge_head[maxn],N,M,F[maxn][maxlog+];
 int Dep[maxn],num_treenode=,root[maxn],lsh_cnt=,cor[maxn],U,V,K,X,Y;
 inline int rd(){
     int x=,f=;char c=getchar();
     while(c<''||c>''){if(c=='-')f=-; c=getchar();}
     while(c>=''&&c<=''){x=x*+c-''; c=getchar();}
     return f*x;
 }
 struct Edge{int to,nx;}edge[maxn<<];
 inline void Add_edge(int from,int to){
     edge[++num_edge].nx=edge_head[from];
     edge[num_edge].to=to;
     edge_head[from]=num_edge;
     return;
 }
 struct Tree{int cnt,l,r,ls,rs;}t[(maxn<<)+maxn*maxlog];
 inline void Build(int x,int l,int r){
     t[x].l=l;t[x].r=r;int mid=(l+r)>>;
     if(l==r)return;
     Build(t[x].ls=++num_treenode,l,mid);
     Build(t[x].rs=++num_treenode,mid+,r);
     return;
 }
 inline void Update(int u,int x,int s){
     int l=t[u].l,r=t[u].r,mid=(l+r)>>;
     t[x].l=l;t[x].r=r;
     if(l==r&&l==s){t[x].cnt=t[u].cnt+; return;}
     if(s<=mid){t[x].rs=t[u].rs; Update(t[u].ls,t[x].ls=++num_treenode,s);}
           else{t[x].ls=t[u].ls; Update(t[u].rs,t[x].rs=++num_treenode,s);}
     t[x].cnt=t[t[x].ls].cnt+t[t[x].rs].cnt;
     return;
 }
 struct A_{int yn,hn,id;}A[maxn];
 inline void Dfs(int x,int fa){
     Dep[x]=Dep[fa]+;
     F[x][]=fa;
     for(int i=;i<=maxlog;i++)
         F[x][i]=F[F[x][i-]][i-];
     Update(root[fa],root[x]=++num_treenode,A[x].hn);
     for(int i=edge_head[x];i;i=edge[i].nx){
         int y=edge[i].to;
         if(y==fa)continue;
         Dfs(y,x);
     }
     return;
 }
 inline int LCA(int x,int y){
     if(Dep[x]<Dep[y])swap(x,y);
     for(int i=maxlog;i>=;i--){
         if(Dep[F[x][i]]>=Dep[y])x=F[x][i];
         if(x==y)return x;
     }
     for(int i=maxlog;i>=;i--){
         if(F[x][i]!=F[y][i]){
             x=F[x][i];y=F[y][i];
         }
     }
     return F[x][];
 }
 inline bool cmp(const A_&a,const A_&b){return a.yn<b.yn;}
 inline bool cmp2(const A_&a,const A_&b){return a.id<b.id;}
 inline int Query(int u,int v,int z,int w,int k){
     int l=t[u].l,r=t[u].r;
     if(l==r)return l;
     int a=t[t[u].ls].cnt+t[t[v].ls].cnt-t[t[z].ls].cnt-t[t[w].ls].cnt;
     if(a>=k)return Query(t[u].ls,t[v].ls,t[z].ls,t[w].ls,k);
        else return Query(t[u].rs,t[v].rs,t[z].rs,t[w].rs,k-a);
 }
 int main(){
     N=rd();M=rd();
     for(int i=;i<=N;i++)A[i].yn=rd(),A[i].id=i;
     sort(A+,A+N+,cmp);
     cor[A[].hn=++lsh_cnt]=A[].yn;
     for(int i=;i<=N;i++)
         if(A[i-].yn!=A[i].yn)
             cor[A[i].hn=++lsh_cnt]=A[i].yn;
         else A[i].hn=lsh_cnt;
     sort(A+,A+N+,cmp2);
     for(int i=;i<N;i++){
         X=rd();Y=rd();
         Add_edge(X,Y);Add_edge(Y,X);
     }
     Build(root[]=++num_treenode,,lsh_cnt+);
     Dfs(,);
     while(M--){
         U=rd();V=rd();K=rd();
         U^=lastans;
         int lca=LCA(U,V);
         lastans=Query(root[U],root[V],root[lca],root[F[lca][]],K);
         lastans=cor[lastans];
         if(M==)printf("%d",lastans);else printf("%d\n",lastans);
     }
     return ;
 }

---

By:AlenaNuna