主席树+LCA——Count on a tree

传送门

题解：
找u->v上面的第k大的权值，那么先要把这条链的信息给截取下来，也就是:rt[u]-tr[lca]+rt[v]-rt[fa[lca]]

#include<bits/stdc++.h>
using namespace std;
const int maxn=400005;
const int tp=18;
int n,m;
int root[maxn],lson[maxn*20],rson[maxn*20],tsize[maxn*20];
int ne[maxn<<2],head[maxn<<2],w[maxn],ls[maxn],e[maxn<<2];
int dep[maxn];
int f[maxn][tp];
int tot,cnt,sz;
void add(int a,int b)
{
    ne[cnt]=head[a],e[cnt]=b,head[a]=cnt++;
}
int get_ls(int x)
{
    return lower_bound(ls+1,ls+1+sz,x)-ls;
}
void update(int fa_,int now,int l,int r,int x)
{
    tsize[now]=tsize[fa_]+1;
    lson[now]=lson[fa_];
    rson[now]=rson[fa_];
    if(l==r) return ;
    int mid=l+r>>1;
    if(x<=mid) update(lson[fa_],lson[now]=++tot,l,mid,x);
    else update(rson[fa_],rson[now]=++tot,mid+1,r,x);
}
void dfs(int u,int father)
{
    dep[u]=dep[father]+1;
    f[u][0]=father;
    for(int i=1;i<tp;i++) f[u][i]=f[f[u][i-1]][i-1];
    update(root[father],root[u]=++tot,1,sz,get_ls(w[u]));
    for(int i=head[u];~i;i=ne[i]){
        int j=e[i];
        if(j==father) continue;
        dfs(j,u);
    }
}
int qlca(int a,int b)
{
    if(dep[a]>dep[b]) swap(a,b);
    if(dep[a]<dep[b]){
        int depp=dep[b]-dep[a];
        for(int i=0;i<tp;i++){
            if(depp&(1<<i)) b=f[b][i];
        }
    }
    if(a==b) return a;
    for(int i=tp-1;i>=0;i--){
        if(f[a][i]!=f[b][i]){
            a=f[a][i];
            b=f[b][i];
        }
    }
    return f[a][0];
}
int query(int fa_,int now,int lcanow,int falcanow,int l,int r,int x)
{
    if(l==r) return l;
    int sum=tsize[lson[fa_]]+tsize[lson[now]]-tsize[lson[lcanow]]-tsize[lson[falcanow]];
    int mid=l+r>>1;
    if(sum>=x) query(lson[fa_],lson[now],lson[lcanow],lson[falcanow],l,mid,x);
    else query(rson[fa_],rson[now],rson[lcanow],rson[falcanow],mid+1,r,x-sum);
}
signed main()
{
    memset(head,-1,sizeof head);
    tot=cnt=0;
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++) {
        scanf("%d",&w[i]);
        ls[i]=w[i];
    }
    sort(ls+1,ls+1+n);
    sz=unique(ls+1,ls+1+n)-ls-1;
    int x,y,z;
    for(int i=1;i<n;i++){
        scanf("%d%d",&x,&y);
        add(x,y);add(y,x);
    }
    dfs(1,-1);
    for(int i=1;i<=m;i++){
        scanf("%d%d%d",&x,&y,&z);
        int lca=qlca(x,y);
        printf("%d\n",ls[query(root[x],root[y],root[lca],root[f[lca][0]],1,sz,z)]);
    }
}