题目传送门
。
解法:
树上主席树。
我在想要不要离散化。
想看看数据范围。
翻了翻discuss发现很神奇啊。
什么MLE,什么开longlong??
然后我就很乖的开了long long。
第i棵主席树维护根到i的信息。
如何求x到y路径的信息呢。
p表示lca。
那么用x到根+y到根-p到根-p的父亲到根就得到x到y的信息。
然后作第K大就行了呀。
记得离散化。
代码实现:
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<iostream>
#include<algorithm>
#include<cmath>
#include<queue>
using namespace std;
typedef long long ll;
struct node {int lc,rc,c;}t[5100000];int cnt,rt[110000];
void build(int &u,int l,int r,int p) {
if(u==0)u=++cnt;t[u].c++;
if(l==r)return ;int mid=(l+r)/2;
if(p<=mid)build(t[u].lc,l,mid,p);
else build(t[u].rc,mid+1,r,p);
}
void Merge(int &u1,int u2) {
if(u1==0) {u1=u2;return ;}if(u2==0)return ;
t[u1].c+=t[u2].c;
Merge(t[u1].lc,t[u2].lc);
Merge(t[u1].rc,t[u2].rc);
}
struct trnode {int x,y,next;}a[210000];int len,last[110000];
void ins(int x,int y) {len++;a[len].x=x;a[len].y=y;a[len].next=last[x];last[x]=len;}
int mx[21][110000],dep[110000];
void dfs(int x) {
for(int k=last[x];k;k=a[k].next) {
int y=a[k].y;
if(y!=mx[0][x]) {
mx[0][y]=x;dep[y]=dep[x]+1;
dfs(y);
}
}
}
int bin[21],n;
void work() {
bin[0]=1;for(int i=1;i<=20;i++)bin[i]=bin[i-1]*2;
for(int j=1;j<=20;j++)for(int i=1;i<=n;i++)if(dep[i]>=bin[j])
mx[j][i]=mx[j-1][mx[j-1][i]];
}
void merge(int x) {
for(int k=last[x];k;k=a[k].next) {
int y=a[k].y;
if(y!=mx[0][x]){Merge(rt[y],rt[x]);merge(y);}
}
}ll S[110000];
struct edge {ll x;int id;}s[110000];
bool cmp(edge n1,edge n2) {return n1.x<n2.x;}
int lca(int x,int y) {
if(dep[x]>dep[y])swap(x,y);
for(int i=20;i>=0;i--)if(dep[y]-dep[x]>=bin[i])y=mx[i][y];
if(x==y)return x;
for(int i=20;i>=0;i--)if(mx[i][x]!=mx[i][y]&&dep[x]>=bin[i]) {
x=mx[i][x];y=mx[i][y];
}return mx[0][x];
}
ll find(int u1,int u2,int u3,int u4,int l,int r,int k) {
if(l==r)return S[l];
int mid=(l+r)/2;
int c=t[t[u1].lc].c+t[t[u2].lc].c-t[t[u3].lc].c-t[t[u4].lc].c;
if(c>=k) return find(t[u1].lc,t[u2].lc,t[u3].lc,t[u4].lc,l,mid,k);
else return find(t[u1].rc,t[u2].rc,t[u3].rc,t[u4].rc,mid+1,r,k-c);
}
ll ans[110000];
int main() {
int m;scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++) {scanf("%lld",&s[i].x);s[i].id=i;}
sort(s+1,s+1+n,cmp);int tot=0;
for(int i=1;i<=n;i++) {
if(s[i].x!=s[i-1].x)tot++;S[tot]=s[i].x;
build(rt[s[i].id],1,n,tot);
}len=0;memset(last,0,sizeof(last));
for(int i=1;i<n;i++) {
int x,y;scanf("%d%d",&x,&y);
ins(x,y);ins(y,x);
}mx[0][1]=0;dep[1]=0;dfs(1);work();merge(1);
ll lastans=0ll;
for(int i=1;i<=m;i++) {
int x,y,k;scanf("%d%d%d",&x,&y,&k);x^=lastans;
int Lca=lca(x,y);
lastans=find(rt[x],rt[y],rt[Lca],rt[mx[0][Lca]],1,n,k);
ans[i]=lastans;
}
for(int i=1;i<m;i++)printf("%lld\n",ans[i]);
printf("%lld",ans[m]);
return 0;
}