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
using namespace std;
const int maxx=1e6+5;
int f[maxx][25];
struct node{
    int l,r;
    int sum;
}t[maxx*40];
vectore[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]=0;i--){
        if(d&(1<=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>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>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<