慕名前来做树上路径第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;
}