COGS[2701] 动态树

  只要我们统计出一个点出发所有虚边连接的点的子树大小 x，那么对于每一个查询，先执行access，之后 x+1 即为答案。

  维护方式见网上，我只提供我的板子。

 

// q.c

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
using namespace std;
const int M=200000+10;
/*******************************************************************************************/
int f[M],l[M],r[M],s[M],sv[M]; bool tag[M];
bool is_root(int x) {
	return (l[f[x]]!=x)&&(r[f[x]]!=x);
}
void UP(int x) {
	s[x]=s[l[x]]+s[r[x]]+sv[x]+1;
}
void l_rot(int x) {
	int y,z; y=f[x],z=f[y],f[x]=z;
	if(y==l[z]) l[z]=x;
	if(y==r[z]) r[z]=x;
	if(l[x]) f[l[x]]=y;
	r[y]=l[x],l[x]=y,f[y]=x;
	UP(y),UP(x);
}
void r_rot(int x) {
	int y,z; y=f[x],z=f[y],f[x]=z;
	if(y==l[z]) l[z]=x;
	if(y==r[z]) r[z]=x;
	if(r[x]) f[r[x]]=y;
	l[y]=r[x],r[x]=y,f[y]=x;
	UP(y),UP(x);
}
void DOWN(int x) {
	if(!tag[x]) return ;
	swap(l[x],r[x]),tag[x]^=1;
	tag[l[x]]^=1,tag[r[x]]^=1;
}
void splay(int x) {
	DOWN(x); int y,z; y=f[x],z=f[y];
	while(!is_root(x)) {
		DOWN(z),DOWN(y),DOWN(x);
		if(is_root(y)) x==l[y]?r_rot(x):l_rot(x);
		else if(x==l[y]&&y==l[z]) r_rot(y),r_rot(x);
		else if(x==l[y]&&y==r[z]) r_rot(x),l_rot(x);
		else if(x==r[y]&&y==l[z]) l_rot(x),r_rot(x);
		else l_rot(y),l_rot(x); y=f[x],z=f[y]; // wtf.
	}
}
/*******************************************************************************************/
void access(int x) {
	for(int y=0;x;y=x,x=f[x]) {
		splay(x);
		sv[x]+=s[r[x]]-s[y];
		r[x]=y,UP(x); // 这里要UP.
	}
}
void make_root(int x) {
	access(x),splay(x); tag[x]^=1;
}
void Link(int x,int y) {
	make_root(x);
	access(y),splay(y); f[x]=y,sv[y]+=s[x];
}
void query(int x) {
	access(x),splay(x); printf("%d\n",sv[x]+1);
}
/*******************************************************************************************/
int main() {
	freopen("dynamic_tree.in","r",stdin);
	freopen("dynamic_tree.out","w",stdout);
	int n,m,opt,a,b;
	scanf("%d%d",&n,&m);
	for(int i=1;i<=m;i++) {
		scanf("%d%d",&opt,&a);
		if(opt==1) make_root(a);
		else if(opt==2) query(a);
		else scanf("%d",&b),Link(b,a);
	}
	return 0;
}