luogu P2633 Count on a tree

背景：

今天快船又双战胜了勇士，$\text{Durant}$空砍$45\text{pts}$，$\text{Louis Williams}\text{NB}$。

题目传送门：

https://www.luogu.org/problemnew/show/P2633

题意：

一棵树，在线询问$x-y$路径上的第$k$小值。

思路：

就是用主席树维护一棵树。
具体来说，对于每一个点，就是你维护一棵主席树从当前点到根的值。
你考虑树上差分。
$$an{s}_{x,y}=an{s}_{1,x}+an{s}_{1,y}-an{s}_{\text{1},lcax,y}-an{s}_{1,{\text{fa}}_{\text{lca}x,y}}$$
对于$\text{lca}$你用倍增不就可以可以了吗。
$\text{INF}$的初始化错了，调了我一天。
$\text{over.}$

代码：

#include<cstdio>
#include<cstring>
#include<algorithm>
#define INF (2147483647)
#define LL long long
using namespace std;
	struct node{int x,y,next;} a[200010];
	int last[200010],f[200010][20],dep[200010];
	int ls[8000010],rs[8000010],root[200010],tot[8000010],d[200010];
	int n,m,len=0;
void ins(int x,int y)
{
	a[++len]=(node){x,y,last[x]}; last[x]=len;
}
void dfs1(int x)
{
	for(int i=last[x];i;i=a[i].next)
	{
		int y=a[i].y;
		if(y==f[x][0]) continue;
		f[y][0]=x;
		dep[y]=dep[x]+1;
		dfs1(y);
	}
}
void init()
{
	for(int i=1;i<=18;i++)
		for(int j=1;j<=n;j++)
			f[j][i]=f[f[j][i-1]][i-1];
}
int lca(int x,int y)
{
	if(dep[x]<dep[y]) swap(x,y);
	for(int i=18;i>=0;i--)
		if(dep[f[x][i]]>=dep[y]) x=f[x][i];
	if(x==y) return x;
	for(int i=18;i>=0;i--)
		if(f[x][i]!=f[y][i]) x=f[x][i],y=f[y][i];
	return f[x][0];
}
void change(int last,int &now,int x,int l,int r)
{
	if(!now) now=++len;
	tot[now]=tot[last]+1;
	if(l==r) return;
	int mid=((LL)l+r)/2ll;
	if(x<=mid)
	{
		rs[now]=rs[last];
		change(ls[last],ls[now],x,l,mid);
	}
	else
	{
		ls[now]=ls[last];
		change(rs[last],rs[now],x,mid+1,r);
	}
}
void dfs2(int x)
{
	change(root[f[x][0]],root[x],d[x],-INF,INF);
	for(int i=last[x];i;i=a[i].next)
	{
		int y=a[i].y;
		if(y==f[x][0]) continue;
		dfs2(y);
	}
}
int solve(int x,int y,int u,int v,int k,int l,int r)
{
	if(l==r) return l;
	int tmp=tot[ls[x]]+tot[ls[y]]-tot[ls[u]]-tot[ls[v]],mid=((LL)l+r)/2ll;
	return k<=tmp?solve(ls[x],ls[y],ls[u],ls[v],k,l,mid):solve(rs[x],rs[y],rs[u],rs[v],k-tmp,mid+1,r);
}
int work(int x,int y,int k)
{
	int LCA=lca(x,y);
	return solve(root[x],root[y],root[LCA],root[f[LCA][0]],k,-INF,INF);
}
int main()
{
	int x,y,k,lastans=0;
	scanf("%d %d",&n,&m);
	for(int i=1;i<=n;i++)
		scanf("%d",&d[i]);
	for(int i=1;i<n;i++)
	{
		scanf("%d %d",&x,&y);
		ins(x,y),ins(y,x);
	}
	dep[1]=1;
	dfs1(1);
	init();
	len=0;
	dfs2(1);
	for(int i=1;i<=m;i++)
	{
		scanf("%d %d %d",&x,&y,&k);
		x^=lastans;
		printf("%d\n",lastans=work(x,y,k));
	}
}