Count on a tree（主席树）

题目描述
给定一棵 n个节点的树，每个点有一个权值。有 m 个询问，每次给你 u,v,k，你需要回答 u^ last 和 v 这两个节点间第 k 小的点权。

其中 last 是上一个询问的答案，定义其初始为 0，即第一个询问的 u是明文。

输入格式
第一行两个整数 n,m。

第二行有 n 个整数，其中第 i 个整数表示点 i的权值。

后面 n-1 行每行两个整数 x,y，表示点 x 到点 y 有一条边。

最后 m 行每行两个整数 u,v,k，表示一组询问。

输出格式
m 行，每行一个正整数表示每个询问的答案。
————————————————————————————
题目要求第k小的点权，于是我们想到了主席树，但一般的主席树是针对数组的，而本题是棵树，但无所谓，dfs时相仿数组时i到i+1变为fa到u，然后考虑如何去求第k小，我们主席树已经存好了每个节点时的状况，然后再类比树上前缀和的操作，用u点的主席树+v点的主席树-lca(u,v)的主席树-lca(u,v)父节点的主席树，在这样产生的主席树上查找第k小的排名，最后输出它原来的点权。
如图

理解上式后，问题基本可以解决了。另外注意离散化和主席树的代码细节。代码如下：

#include<bits/stdc++.h>
using namespace std;
const int N=5e6;
#define int long long
int n,m,index_,a[N],b[N],dep[N],len,lg[N],f[N][24],ans;
int val[N],L[N],R[N],rt[N],cnt,x,y,z,k,first[N],Next[N],to[N];
int Read()
{
	int x=0,f=1;
	char ch=getchar();
	while(!isdigit(ch))
	{
		if(ch=='-') f=-1;
		ch=getchar();
	}
	while(isdigit(ch))
	{
		x=(x<<1)+(x<<3)+ch-'0';
		ch=getchar();
	}
	return f*x;
}
void add(int a,int b)
{
	index_++;
	Next[index_]=first[a];
	first[a]=index_;
	to[index_]=b;
}
int build(int l,int r)
{
	cnt++;
	int u=cnt;
	if(l==r) return u;
	int mid=(l+r)>>1;
	L[u]=build(l,mid);
	R[u]=build(mid+1,r);
	return u;
}
int update(int l,int r,int pre,int X)
{
	cnt++;
	int u=cnt;
	val[u]=val[pre]+1;
	L[u]=L[pre],R[u]=R[pre];
	if(l==r) return u;
	int mid=(l+r)>>1;
	if(X<=mid) L[u]=update(l,mid,L[pre],X);
	else R[u]=update(mid+1,r,R[pre],X);
	return u;
}
void dfs1(int u,int fa)
{		
    rt[u]=update(1,len,rt[fa],a[u]);
	f[u][0]=fa;
	dep[u]=dep[fa]+1;
	for(int i=1;i<=lg[dep[u]];i++) f[u][i]=f[f[u][i-1]][i-1];
	for(int i=first[u];i;i=Next[i])
	{
		int v=to[i];
		if(v==fa) continue;
        dfs1(v,u);
	}
}
int lca(int u,int v)
{
	if(dep[u]<dep[v]) swap(u,v);
	while(dep[u]>dep[v]) u=f[u][lg[dep[u]-dep[v]]-1];
	if(u==v) return u;
	for(int i=lg[dep[u]]-1;i>=0;i--)
	if(f[u][i]!=f[v][i]) u=f[u][i],v=f[v][i];
	return f[u][0];
}
int query(int l,int r,int now1,int now2,int pre1,int pre2,int K)
{
	if(l==r) return l;
	int mid=(l+r)>>1;
	int zhi=val[L[now1]]+val[L[now2]]-val[L[pre1]]-val[L[pre2]];
	if(K<=zhi) return query(l,mid,L[now1],L[now2],L[pre1],L[pre2],K);
	else return query(mid+1,r,R[now1],R[now2],R[pre1],R[pre2],K-zhi);
}
signed main()
{
	n=Read(),m=Read();
	for(int i=1;i<=n;i++) lg[i]=lg[i-1]+(1<<lg[i-1]==i);
	for(int i=1;i<=n;i++) a[i]=Read(),b[i]=a[i];
	sort(b+1,b+1+n);
	len=unique(b+1,b+1+n)-b-1;
	for(int i=1;i<=n;i++)
	a[i]=lower_bound(b+1,b+1+len,a[i])-b;
	rt[0]=build(1,len);
	for(int i=1;i<n;i++) x=Read(),y=Read(),add(x,y),add(y,x);
	dfs1(1,0);
	ans=0;
	for(int i=1;i<=m;i++)
	{
		x=Read(),y=Read(),k=Read();
		x^=ans,z=lca(x,y);
		ans=b[query(1,len,rt[x],rt[y],rt[z],rt[f[z][0]],k)];
		printf("%lld\n",ans);
	}
	return 0;
}