SPOJ COT Count on a tree 树上主席树&链上点权第k大

题意：

给定一棵N个节点的树，每个点有一个权值，对于M个询问(u,v,k)，输出节点u到节点v这条链上第k大的点。

思路：

树上主席树，对于树上每一个节点建一个权值线段树，利用lca求出u-->lca-->v这条链，对于这几颗线段树root[u],root[v],root[lca],root[f[lca][0]]，利用容斥计算出这条链上的权值线段树，再进行查询。

#include<bits/stdc++.h>
using namespace std;
const int maxx=1e6+5;
int f[maxx][25];
struct node{
    int l,r;
    int sum;
}t[maxx*40];
vector<int>e[maxx];
int a[maxx],dep[maxx],rk[maxx],vis[maxx],root[maxx];
int len,n,m,cnt=0,last=0;
int LCA(int u,int v){
    if(dep[u]<dep[v]){
        swap(u,v);
    }
    int d=dep[u]-dep[v];
    for(int i=20;i>=0;i--){
        if(d&(1<<i))
            u=f[u][i];
    }
    if(u==v)return u;
    for(int i=20;i>=0;i--){
        if(f[u][i]!=f[v][i]){
            u=f[u][i];
            v=f[v][i];
        }
    }
    return f[u][0];
}

void update(int l,int r,int &cur,int pre,int k){
    t[++cnt]=t[pre];
    t[cnt].sum++;
    cur=cnt;
    if(l==r)return;
    int mid=(l+r)/2;
    if(k<=mid) update(l,mid,t[cur].l,t[pre].l,k);
    else update(mid+1,r,t[cur].r,t[pre].r,k);
}
void dfs(int x,int fa,int d){
    int k=lower_bound(rk+1,rk+len+1,a[x])-rk;
    vis[x]=1;dep[x]=d;
    update(1,n,root[x],root[fa],k);
    for(int i=0;i<e[x].size();i++){
        int j=e[x][i];
        if(!vis[j]){
            f[j][0]=x;
            dfs(j,x,d+1);
        }

    }
}
int query(int l,int r,int rt1,int rt2,int rt3,int rt4,int k){
    int sum=t[t[rt1].l].sum+t[t[rt2].l].sum-t[t[rt3].l].sum-t[t[rt4].l].sum;
    if(l==r)return l;
    int mid=(l+r)/2;
    if(k<=sum)return query(l,mid,t[rt1].l,t[rt2].l,t[rt3].l,t[rt4].l,k);
    else return query(mid+1,r,t[rt1].r,t[rt2].r,t[rt3].r,t[rt4].r,k-sum);
}
int main(){
    ios::sync_with_stdio(0);
    cin.tie(0);cout.tie(0);
    cin>>n>>m;
    for(int i=1;i<=n;i++){
        cin>>a[i];
        rk[i]=a[i];
    }
    sort(rk+1,rk+n+1);
    len=unique(rk+1,rk+n+1)-rk-1;
    for(int i=1;i<=n-1;i++){
        int u,v;
        cin>>u>>v;
        e[u].push_back(v);
        e[v].push_back(u);
    }
    dfs(1,0,1);
    for(int j=1;j<=20;j++){
        for(int i=1;i<=n;i++){
            f[i][j]=f[f[i][j-1]][j-1];
        }
    }
    for(int i=0;i<m;i++){
        int u,v,k;
        cin>>u>>v>>k;
        int lca=LCA(u,v);
        int ans=query(1,n,root[u],root[v],root[lca],root[f[lca][0]],k);
        cout<<rk[ans]<<endl;
    }
}