BZOJ 2588: Count on a tree

加个dfs序就可以把静态kth移到树上来做。树上差分一下即可。

顺便再多写一个dfs用以树剖求lca。

#include <bits/stdc++.h>
using namespace std;
const int N=1e5+5;
int n,q,m,u,v,t,LCA,cnt,last;
int ll[N<<5],rr[N<<5],sum[N<<5],T[N];
int a[N],b[N];
int f[N],size[N],d[N],son[N],top[N];
int cntt,head[N];
struct edge{int next,to;}e[N<<1];

inline void add(int u,int v)
{
	cntt++;
	e[cntt].next=head[u];
	e[cntt].to=v;
	head[u]=cntt;
}

int change(int pre,int l,int r,int x)
{
	int rt=++cnt;
	ll[rt]=ll[pre]; rr[rt]=rr[pre]; sum[rt]=sum[pre]+1;
	if (l<r)
	{
		int mid=l+r>>1;
		if (x<=mid) ll[rt]=change(ll[pre],l,mid,x);
		else rr[rt]=change(rr[pre],mid+1,r,x);  
	}
	return rt;
}

void dfs(int u,int fa)
{
	size[u]=1;
	int t=lower_bound(a+1,a+m+1,b[u])-a;
	T[u]=change(T[fa],1,m,t);
	for (register int i=head[u]; i; i=e[i].next)
	if (e[i].to!=fa)
	{
		f[e[i].to]=u; d[e[i].to]=d[u]+1;
		dfs(e[i].to,u);
		size[u]+=size[e[i].to];
		if (size[e[i].to]>size[son[u]]) son[u]=e[i].to;
	}
}

void dfs2(int u,int TP)
{
	top[u]=TP;
	if (son[u]) dfs2(son[u],TP);
	for (register int i=head[u]; i; i=e[i].next) 
	if (e[i].to!=f[u] && e[i].to!=son[u]) dfs2(e[i].to,e[i].to);
}

inline int lca(int u,int v)
{
	while (top[u]!=top[v])
	{
		if (d[top[u]]<d[top[v]]) swap(u,v);
		u=f[top[u]];
	}
	if (d[u]>d[v]) swap(u,v);
	return u;	
}

int query(int u,int v,int LCA,int fa,int l,int r,int x)
{
	if (l==r) return l;
	int now=sum[ll[u]]+sum[ll[v]]-sum[ll[LCA]]-sum[ll[fa]];
	int mid=l+r>>1;
	if (x<=now) return query(ll[u],ll[v],ll[LCA],ll[fa],l,mid,x);
	else return query(rr[u],rr[v],rr[LCA],rr[fa],mid+1,r,x-now);
}

int main(){
	scanf("%d%d",&n,&q);
	for (register int i=1; i<=n; ++i) scanf("%d",&a[i]),b[i]=a[i];
	sort(a+1,a+n+1);
	m=unique(a+1,a+n+1)-a-1;
	
	for (register int i=1; i<n; ++i)
	{
		scanf("%d%d",&u,&v);
		add(u,v); add(v,u);
	}
	
	dfs(1,0); dfs2(1,1);
	
	while (q--)
	{
		scanf("%d%d%d",&u,&v,&t);
		u^=last;
		LCA=lca(u,v);
		t=query(T[u],T[v],T[LCA],T[f[LCA]],1,m,t);
		printf("%d\n",a[t]);
		last=a[t];
	}
return 0;
}