BZOJ 3123 [SDO12013] 森林 lca+主席树+启发式合并

给出一个$n\le 8e4$的有点权的森林，然后有$q\le 8e4$个操作，分别是查询路径$u$和$v$路径上的第$k$小的权值和连接$x$和$y$两个点。
路径上$k$小权值只要在dfs的时候建主席树就可以了。合并的时候用并查集维护每个树的大小，把小的往大的上面合并。
时间复杂度$O(nlo{g}^{2}n)$

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int inf=0x3f3f3f3f;
const ll INF=LONG_LONG_MAX;
const int N=8e4+7;
int a[N],b[N],Fa[N][19],dep[N];
vector<int> G[N];
int fa[N],sz[N];

int n,m,q;
int ans=0;
int tot=0;

int rt[N],ls[500*N],rs[500*N],cnt[500*N],tag=0;

void clear() {
	for(int i=1;i<=n;i++) 
	 	G[i].clear();
	tot=ans=tag=0;
	memset(Fa,0,sizeof(Fa));
	memset(dep,0,sizeof(dep)); 
}
void update(int pre,int &o,int l,int r,int k) {
	o=++tag;
	cnt[o]=cnt[pre]+1;
	ls[o]=ls[pre];
	rs[o]=rs[pre];
	if(l==r) return;
	int mid=(l+r)>>1;
	if(k<=mid) update(ls[pre],ls[o],l,mid,k);
	else update(rs[pre],rs[o],mid+1,r,k);
}
int query(int u,int v,int lca1,int lca2,int l,int r,int k) {
	if(l==r) return l;
	int mid=(l+r)>>1;
	int sum=cnt[ls[u]]+cnt[ls[v]]-cnt[ls[lca1]]-cnt[ls[lca2]];
	if(k<=sum) return query(ls[u],ls[v],ls[lca1],ls[lca2],l,mid,k);
	else return query(rs[u],rs[v],rs[lca1],rs[lca2],mid+1,r,k-sum);
}
void dfs(int u,int f,int r) { 
	Fa[u][0]=f;	
	fa[u]=r;
	sz[r]++;
	dep[u]=dep[f]+1;
	for(int i=1;i<=18;i++)
		Fa[u][i]=Fa[Fa[u][i-1]][i-1];
	update(rt[Fa[u][0]],rt[u],1,tot,a[u]);
	for(int i=0;i<(int)G[u].size();i++) {
		int v=G[u][i]; 
		if(v==f) continue;
		dfs(v,u,r);
	}
}
int lca(int x,int y) {
	if(dep[x]<dep[y]) swap(x,y);
	for(int i=18;i>=0;i--) {
		if((1<<i)<=dep[x]-dep[y]) {
			x=Fa[x][i];
		}
	}
	if(x==y) return x;
	for(int i=18;i>=0;i--) {
		if(Fa[x][i]!=Fa[y][i]) {
			x=Fa[x][i];
			y=Fa[y][i];
		}
	}
	return Fa[x][0];
}
int find(int x) {
	if(x==fa[x]) return x;
	else return fa[x]=find(fa[x]);
}
void link(int x,int y) {
	G[x].push_back(y);
	G[y].push_back(x);
	int fx=find(x);
	int fy=find(y);
	if(sz[fx]<sz[fy]) dfs(x,y,fy);
	else dfs(y,x,fx);
}
int main() {
	int T;
	scanf("%d",&T);
	T=1;
	while(T--) {
		scanf("%d%d%d",&n,&m,&q);	
		for(int i=1;i<=n;i++) {
			scanf("%d",&a[i]);
			b[i]=a[i];
			fa[i]=i;
			sz[i]=1;
		}
		sort(b+1,b+1+n);
		tot=unique(b+1,b+1+n)-(b+1);
		for(int i=1;i<=n;i++) {
			a[i]=lower_bound(b+1,b+1+tot,a[i])-b;
		}
		for(int i=1;i<=m;i++) {
			int u,v;
			scanf("%d%d",&u,&v);
			G[u].push_back(v);
			G[v].push_back(u);
		}
		for(int i=1;i<=n;i++) {
			if(fa[i]==i) {
				dfs(i,0,i);
			}
		}
		while(q--) {
			char opt[5];
			scanf("%s",opt);
			if(opt[0]=='Q') {
				int x,y,k;
				scanf("%d%d%d",&x,&y,&k);
				x^=ans,y^=ans,k^=ans;
				int lca1=lca(x,y);
				int lca2=Fa[lca1][0];			
				int id=query(rt[x],rt[y],rt[lca1],rt[lca2],1,tot,k);
				printf("%d\n",b[id]);	
				ans=b[id]; 		
			}
			else if(opt[0]=='L') {
				int x,y;
				scanf("%d%d",&x,&y);
				x^=ans,y^=ans;
				link(x,y); 
			}
		}
	}
	return 0;
}