BZOJ_3123_[Sdoi2013]森林_主席树+启发式合并

BZOJ_3123_[Sdoi2013]森林_主席树+启发式合并

Description

Input

第一行包含一个正整数testcase，表示当前测试数据的测试点编号。保证1≤testcase≤20。
第二行包含三个整数N，M，T，分别表示节点数、初始边数、操作数。第三行包含N个非负整数表示 N个节点上的权值。
 接下来 M行，每行包含两个整数x和 y，表示初始的时候，点x和点y 之间有一条无向边, 接下来 T行，每行描述一个操作，格式为“Q x y k”或者“L x y ”，其含义见题目描述部分。

Output

对于每一个第一类操作，输出一个非负整数表示答案。

Sample Input

1
8 4 8
1 1 2 2 3 3 4 4
4 7
1 8
2 4
2 1
Q 8 7 3 Q 3 5 1
Q 10 0 0
L 5 4
L 3 2 L 0 7
Q 9 2 5 Q 6 1 6

Sample Output

2
2
1
4
2

HINT

对于第一个操作 Q 8 7 3，此时 lastans=0，所以真实操作为Q 8^0 7^0 3^0，也即Q 8 7
3。点8到点7的路径上一共有5个点，其权值为4 1 1 2 4。这些权值中，第三小的为 2，输出 2，lastans变为2。对于第二个操作 Q 3
5 1 ，此时lastans=2，所以真实操作为Q 3^2 5^2 1^2 ，也即Q 1 7 3。点1到点7的路径上一共有4个点，其权值为 1
1 2 4 。这些权值中，第三小的为2，输出2，lastans变为 2。之后的操作类似。

---

考虑没有连边操作怎么做，直接树上主席树就可以。

发现没有删边操作，可以用一个实用的方法：启发式合并。

就是每次把小的那边暴力一个一个拆到大的那边。

实现起来还挺简单的。

代码：

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cstdlib>
#include <cmath>
using namespace std;
#define N 80050
int head[N],to[N<<1],nxt[N<<1],cnt,root[N],bl[N],f[20][N],Lg[N],n,m,val[N],V[N],dep[N],tot,vv;
int t[N*250],ls[N*250],rs[N*250],siz[N];
inline void add(int u,int v) {
	to[++cnt]=v; nxt[cnt]=head[u]; head[u]=cnt;
}
void insert(int l,int r,int v,int &y,int x) {
	if(!y) y=++tot; t[y]=t[x]+1;
	if(l==r) return ;
	int mid=(l+r)>>1;
	if(v<=mid) rs[y]=rs[x],insert(l,mid,v,ls[y],ls[x]);
	else ls[y]=ls[x],insert(mid+1,r,v,rs[y],rs[x]);
}
int qk(int l,int r,int a,int b,int c,int d,int k) {
	if(l==r) return V[l];
	int mid=(l+r)>>1;
	int sizls=t[ls[a]]+t[ls[b]]-t[ls[c]]-t[ls[d]];
	if(k<=sizls) return qk(l,mid,ls[a],ls[b],ls[c],ls[d],k);
	else return qk(mid+1,r,rs[a],rs[b],rs[c],rs[d],k-sizls);
}
void dfs(int x,int y,int anc) {
	dep[x]=dep[y]+1;
	siz[anc]++;
	bl[x]=anc;
	f[0][x]=y;
	int i;
	insert(1,vv,val[x],root[x],root[y]);
	for(i=1;i<=19;i++) f[i][x]=f[i-1][f[i-1][x]];
	for(i=head[x];i;i=nxt[i]) {
		if(to[i]!=y) {
			dfs(to[i],x,anc);
		}
	}
}
int lca(int x,int y) {
	int i;
	if(dep[x]<dep[y]) swap(x,y);
	for(i=Lg[n];i>=0;i--) if(f[i][x]&&dep[f[i][x]]>=dep[y]) x=f[i][x];
	if(x==y) return x;
	for(i=Lg[n];i>=0;i--) if(f[i][x]&&f[i][x]!=f[i][y]) x=f[i][x],y=f[i][y];
	return f[0][x];
}
void solve() {
	// memset(t,0,sizeof(t));
	// memset(ls,0,sizeof(ls));
	// memset(rs,0,sizeof(rs));
	// memset(head,0,sizeof(head)); cnt=0;
	// memset(dep,0,sizeof(dep));
	int i;
	int Q;
	scanf("%d%d%d",&n,&m,&Q);
	for(Lg[0]=-1,i=1;i<=n;i++) Lg[i]=Lg[i>>1]+1;
	int x,y;
	for(i=1;i<=n;i++) scanf("%d",&val[i]),V[i]=val[i];
	sort(V+1,V+n+1);
	vv=unique(V+1,V+n+1)-V-1;
	for(i=1;i<=n;i++) val[i]=lower_bound(V+1,V+vv+1,val[i])-V;
	for(i=1;i<=m;i++) {
		scanf("%d%d",&x,&y); add(x,y); add(y,x);
	}
	for(i=1;i<=n;i++) {
		if(!f[0][i]) dfs(i,0,i);
	}
	int k,ans=0;
	char opt[3];
	while(Q--) {
		scanf("%s%d%d",opt,&x,&y);
		x^=ans; y^=ans;
		if(opt[0]=='Q') {
			scanf("%d",&k); k^=ans;
			int l=lca(x,y);
			// printf("%d\n",l);
			printf("%d\n",ans=qk(1,vv,root[x],root[y],root[l],root[f[0][l]]?root[f[0][l]]:0,k));
		}else {
			if(siz[bl[x]]>siz[bl[y]]) swap(x,y);
			f[0][x]=y; dep[x]=dep[y]+1;
			dfs(x,y,bl[y]); add(x,y); add(y,x);
		}
	}
}
int main() {
	scanf("%*d");
	solve();
}
/*
1
8 4 8
1 1 2 2 3 3 4 4
4 7
1 8
2 4
2 1
Q 8 7 3
Q 3 5 1
Q 10 0 0
L 5 4
L 3 2 L 0 7
Q 9 2 5
Q 6 1 6
*/