【LCA+主席树】SPOJ - COT - Count on a tree

题目链接<https://cn.vjudge.net/problem/SPOJ-COT>

---

题意：

在一棵树上每个点都有一个权值，有若干次询问，每次询问一条简单路上的权值第k小。

---

题解：

主席树主要利用前缀和的性质来确定范围内的值，并用线段树得出答案。本题是建立在树形结构上，所以应当想办法在树上利用前缀和求出范围的权值线段树。

很容易想到跟lca有关。如果我们dfs树，并对树上的每条链做一个主席树，那么简单路上的权值应该为t[u]+t[v]-t[lca(u,v)]-t[fa[lca(u,v)]]。

---

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=1e5+7;

int val[N];
int fa[N][50],d[N];
int p[N],edn;
int n,m,c;

int b[N],cnt;

struct Edge{
    int u,v,nxt;
}edge[N*2];
void add(int u,int v){
    edge[++edn]=(Edge){u,v,p[u]};p[u]=edn;
    edge[++edn]=(Edge){v,u,p[v]};p[v]=edn;
}
int rt[N],t[N*40],ls[N*40],rs[N*40],tot;
int bd(int l,int r){
    int now=++tot;
    if(l==r) return now;
    int m=l+r>>1;
    ls[now]=bd(l,m);
    rs[now]=bd(m+1,r);
    return now;
}
int upd(int o,int l,int r,int p){
    int now=++tot;
    t[now]=t[o]+1;
    if(l==r) return now;
    ls[now]=ls[o];rs[now]=rs[o];
    int m=l+r>>1;
    if(p<=m) ls[now]=upd(ls[o],l,m,p);
    else rs[now]=upd(rs[o],m+1,r,p);
    return now;
}
void dfs(int u,int f){
    rt[u]=upd(rt[f],1,cnt,val[u]);
    for(int i=p[u];~i;i=edge[i].nxt){
        int v=edge[i].v;
        if(v==fa[u][0])continue;
        d[v]=d[u]+1;
        fa[v][0]=u;
        dfs(v,u);
    }
}
void init(){
    for(int j=1;(1<<j)<=n;j++){
        for(int i=1;i<=n;i++){
            fa[i][j]=fa[fa[i][j-1]][j-1];
        }
    }
}
int lca(int a,int b){
    if(d[a]>d[b])swap(a,b);
    int f=d[b]-d[a];
    for(int i=0;i<=30;i++){
        if((1<<i)&f) b=fa[b][i];
    }
    if(a!=b){
        for(int i=30;i>=0;i--){
            if(fa[a][i]!=fa[b][i]){
                a=fa[a][i]; b=fa[b][i];
            }
        }
        a=fa[a][0];
    }
    return a;
}
int que(int u,int v,int o,int oo,int l,int r,int k){
    if(l==r) return l;
    int num=t[ls[u]]+t[ls[v]]-t[ls[o]]-t[ls[oo]];
    int m=l+r>>1;
    if(num>=k) return que(ls[u],ls[v],ls[o],ls[oo],l,m,k);
    else return que(rs[u],rs[v],rs[o],rs[oo],m+1,r,k-num);
}

int main()
{
    scanf("%d%d",&n,&m);
    memset(p,-1,sizeof(p)),edn=-1;
    int u,v,k;
    for(int i=1;i<=n;i++){
        scanf("%d",&val[i]);
        b[i]=val[i];
    }
    sort(b+1,b+1+n);
    cnt=unique(b+1,b+1+n)-b-1;
    for(int i=1;i<=n;i++){
        val[i]=lower_bound(b+1,b+1+cnt,val[i])-b;
    }
    rt[0]=bd(1,cnt);
    for(int i=1;i<n;i++){
        scanf("%d%d",&u,&v);
        add(u,v);
    }
    dfs(1,0);
    init();
    while(m--){
        scanf("%d%d%d",&u,&v,&k);
        int pp=lca(u,v);
        printf("%d\n",b[que(rt[u],rt[v],rt[pp],rt[fa[pp][0]],1,cnt,k)]);
    }
}