BZOJ 2588: Spoj 10628. Count on a tree

慕名前来做树上路径第K大 （为什么我之前没听说过这个模板题

竟然1A了 只是你不知道我是先做了下一题

对于树上每个点 都维护一棵主席树 表示它到根这条链上的全部信息
询问的时候 只要用 x+y-lca-fa[lca]就能得出树上路径的信息了
然后在主席树上减一下 看去左孩子还是右孩子就好啦

主席树又短又快 妙啊
为什么网上面都看不到树剖的题解了…（有的话 给我看看？

#include<bits/stdc++.h>
using namespace std;
const int N=1e5+2;
char B[1<<14],*S=B,*T=B;
#define gc (S==T&&(T=(S=B)+fread(B,1,1<<14,stdin),S==T)?-1:*S++)
inline int read(){
    int x=0,f=1; char ch=gc;
    while(ch<'0' || ch>'9'){if(ch=='-')f=-1; ch=gc;}
    while(ch>='0' && ch<='9'){x=(x<<1)+(x<<3)+ch-'0'; ch=gc;}
    return x*f;
}
struct ED{int y,nex;}e[N<<1]; int len,fir[N];
void ins(int x,int y){
    e[++len]=(ED){y,fir[x]},fir[x]=len;
}
int z[N],f[N][17],a[N],zl,dep[N],tot;
int rt[N],lc[N*20],rc[N*20],s[N*20],L[2],R[2];
void link(int &x,int l,int r,int k){
    lc[++tot]=lc[x],rc[tot]=rc[x],s[tot]=s[x]+1,x=tot;
    if(l==r)return;
    int mid=l+r>>1;
    if(k<=mid)link(lc[x],l,mid,k); else link(rc[x],mid+1,r,k);
}
int query(int l,int r,int k){
    if(l==r)return l;
    int u=0,mid=l+r>>1;
    for(int i=0;i<2;++i)u+=s[lc[R[i]]]-s[lc[L[i]]];
    if(u>=k){
        for(int i=0;i<2;++i) L[i]=lc[L[i]],R[i]=lc[R[i]];
        return query(l,mid,k);
    }
    for(int i=0;i<2;++i) L[i]=rc[L[i]],R[i]=rc[R[i]];
    return query(mid+1,r,k-u);
}
void dfs(int x){
    for(int i=1;i<17;++i)
        if(f[x][i-1])f[x][i]=f[f[x][i-1]][i-1];
        else break;
    dep[x]=dep[f[x][0]]+1;
    rt[x]=rt[f[x][0]],link(rt[x],1,zl,a[x]);
    for(int k=fir[x];k;k=e[k].nex) if(f[x][0]!=e[k].y){
        f[e[k].y][0]=x; dfs(e[k].y);
    }
}
int lca(int x,int y){
    if(dep[x]<dep[y]) x^=y^=x^=y;
    for(int i=16;~i;--i) if(dep[f[x][i]]>=dep[y]) x=f[x][i];
    if(x==y)return x;
    for(int i=16;~i;--i) if(f[x][i]!=f[y][i]) x=f[x][i],y=f[y][i];
    return f[x][0];
}
int main(){
    int i,n=read(),m=read();
    for(i=1;i<=n;++i)a[i]=z[++zl]=read();
    sort(z+1,z+1+zl); zl=unique(z+1,z+1+zl)-z-1;
    for(i=1;i<=n;++i) a[i]=lower_bound(z+1,z+1+zl,a[i])-z;
    for(i=1;i<n;++i){
        int x=read(),y=read();
        ins(x,y),ins(y,x);
    }
    dfs(1);
    int la=0;
    while(m--){
        int u=read()^la,v=read(),l=lca(u,v),k=read();
        R[0]=rt[u],R[1]=rt[v];
        L[0]=rt[l],L[1]=rt[f[l][0]];
        int o=query(1,zl,k); la=z[o];
        printf("%d",z[o]); if(m)puts("");
    }
    return 0;
}