Description
给定一棵N个节点的树,每个点有一个权值,对于M个询问(u,v,k),你需要回答u xor lastans和v这两个节点间第K小的点权。其中lastans是上一个询问的答案,初始为0,即第一个询问的u是明文。
N,M<=100000
强制在线
Solution
可持久化线段树不仅可以用在序列上,还可以在树上做。具体操作就是以每个点的父亲作为上一版本建新的树
考虑查询的两点为x,y,若lca(x,y)=x那么查询root[y]-root[fa[x]],这点与序列上无异。推广一下就是任意两点x,y的查询都相当于查询root[x]-root[lca]+root[y]-root[lca]+root[lca]-root[fa[lca]]
几天不敲可持久化发现不太会了,一开始re是数组没开够
Code
#include <stdio.h>
#include <string.h>
#include <algorithm>
#define rep(i,st,ed) for (int i=st;i<=ed;++i)
#define drp(i,st,ed) for (int i=st;i>=ed;--i)
#define fill(x,t) memset(x,t,sizeof(x))
const int N=200005;
const int E=400005;
struct treeNode{int l,r,c;}t[N*21];
struct edge{int x,y,next;}e[E];
int a[N],b[N],rank[N];
int fa[N][21],dep[N];
int ls[N],edCnt=0;
int root[N],cnt=0;
int lastans=0,size,n,m;
int read() {
int x=0,v=1; char ch=getchar();
for (;ch<'0'||ch>'9';v=(ch=='-')?(-1):(v),ch=getchar());
for (;ch<='9'&&ch>='0';x=x*10+ch-'0',ch=getchar());
return x*v;
}
void addEdge(int x,int y) {
e[++edCnt]=(edge){x,y,ls[x]}; ls[x]=edCnt;
e[++edCnt]=(edge){y,x,ls[y]}; ls[y]=edCnt;
}
int query(int x,int y,int lca,int lcaf,int l,int r,int k) {
if (l==r) return l;
int c=t[t[x].l].c+t[t[y].l].c-t[t[lca].l].c-t[t[lcaf].l].c,mid=(l+r)>>1;
if (c>=k) return query(t[x].l,t[y].l,t[lca].l,t[lcaf].l,l,mid,k);
else return query(t[x].r,t[y].r,t[lca].r,t[lcaf].r,mid+1,r,k-c);
}
void modify(int &now,int pre,int l,int r,int pos) {
now=++cnt;
t[now]=(treeNode){t[pre].l,t[pre].r,t[pre].c+1};
if (l==r) return ;
int mid=(l+r)>>1;
if (pos<=mid) modify(t[now].l,t[pre].l,l,mid,pos);
else modify(t[now].r,t[pre].r,mid+1,r,pos);
}
void buildTree(int &now,int l,int r) {
now=++cnt;
t[now]=(treeNode){0,0,0};
if (l==r) return ;
int mid=(l+r)>>1;
buildTree(t[now].l,l,mid);
buildTree(t[now].r,mid+1,r);
}
void dfs(int now) {
modify(root[now],root[fa[now][0]],1,size,rank[now]);
rep(i,1,20) fa[now][i]=fa[fa[now][i-1]][i-1];
for (int i=ls[now];i;i=e[i].next) {
if (e[i].y==fa[now][0]) continue;
dep[e[i].y]=dep[now]+1;
fa[e[i].y][0]=now;
dfs(e[i].y);
}
}
int get_lca(int x,int y) {
if (dep[x]<dep[y]) std:: swap(x,y);
drp(i,20,0) if (dep[fa[x][i]]>=dep[y]) x=fa[x][i];
if (x==y) return x;
drp(i,20,0) if (fa[x][i]!=fa[y][i]) x=fa[x][i],y=fa[y][i];
return fa[x][0];
}
void solve(int x,int y,int k) {
int lca=get_lca(x,y);
lastans=query(root[x],root[y],root[lca],root[fa[lca][0]],1,size,k);
printf("%d\n", b[lastans]);
}
int main(void) {
n=read(),m=read();
rep(i,1,n) a[i]=b[i]=read();
std:: sort(b+1,b+n+1);
size=std:: unique(b+1,b+n+1)-b-1;
rep(i,1,n) rank[i]=std:: lower_bound(b+1,b+size+1,a[i])-b;
rep(i,2,n) {
int x=read(),y=read();
addEdge(x,y);
}
buildTree(root[0],1,size);
dep[1]=1;
dfs(1);
while (m--) {
int x=read(),y=read(),k=read();
x^=b[lastans];
solve(x,y,k);
}
return 0;
}