[BZOJ3637][LCT]Qtree6

BZOJ3637

LCT维护连通块的一个题
一个简单的想法是直接两个LCT分别维护黑白森林，改变点的时候就把所有出边linkcut一下
然而菊花图可以随便卡
被卡的关键在于这个算法是以点的出度为时间复杂度瓶颈的，所以考虑一个和点的出度无关的算法
树有一个性质，每个点可以有很多个儿子（上面那个算法被卡的关键），但是只有一个父亲（下面这个算法的实现基础），所以我们link cut只考虑一个点和其父亲的关系，复杂度就正确了
询问就输出右子树大小，pushup就和树点涂色差不多，讨论旧实儿子和新实儿子

Code：

#include<bits/stdc++.h>
using namespace std;
inline int read(){
	int res=0,f=1;char ch=getchar();
	while(!isdigit(ch)) {if(ch=='-') f=-f;ch=getchar();}
	while(isdigit(ch)) {res=(res<<1)+(res<<3)+(ch^48);ch=getchar();}
	return res*f;
}
const int N=1e5+5;
int f[N],vis[N<<1],col[N],head[N<<1],nxt[N<<1],tot=0;
inline void add(int x,int y){vis[++tot]=y;nxt[tot]=head[x];head[x]=tot;}
struct LCT{
	int ls[N],rs[N],siz[N],fa[N],s[N];
	LCT(){for(int i=1;i<N;i++) siz[i]=1;}
	inline int isrs(int x){return rs[fa[x]]==x;}
	inline bool isroot(int x){
		if(!x) return true;
		return ls[fa[x]]!=x && rs[fa[x]]!=x;
	}
	inline void pushup(int x){siz[x]=siz[ls[x]]+siz[rs[x]]+1+s[x];}
	inline void rotate(int x){
		int y=fa[x],z=fa[y],b=(ls[y]==x?rs[x]:ls[x]);
		if(z && !isroot(y)) (ls[z]==y?ls[z]:rs[z])=x;
		fa[x]=z,fa[y]=x,b?fa[b]=y:0;
		if(ls[y]==x) rs[x]=y,ls[y]=b;
		else ls[x]=y,rs[y]=b;
		pushup(y);pushup(x);
	}
	inline void splay(int x){
		while(!isroot(x)){
			while(!isroot(fa[x])){
				if(isrs(x)==isrs(fa[x])) rotate(fa[x]);
				else rotate(x);
			}
			rotate(x);
		}
	}
	inline void access(int x){
		for(int y=0;x;y=x,x=fa[x]){
			splay(x);
			s[x]+=siz[rs[x]];
			s[x]-=siz[rs[x]=y];
		}
	}
	inline int findroot(int x){
		access(x);splay(x);
		while(ls[x]) x=ls[x];
		splay(x);return x;
	}
	inline void link(int x){
		splay(x);int y=fa[x]=f[x];
		access(y);splay(y);
		s[y]+=siz[x];siz[y]+=siz[x];
	}
	inline void cut(int x){
		access(x);splay(x);
		ls[x]=fa[ls[x]]=0;
		pushup(x);
	}
}lct[2];
void dfs(int x){
	for(int y,i=head[x];i;i=nxt[i])
		if((y=vis[i])!=f[x])
			f[y]=x,dfs(y),lct[0].link(y);
}
int main(){
	int n,m;
	n=read();
	for(int x,y,i=1;i<n;++i){x=read();y=read();add(x,y);add(y,x);}
	dfs(1);
	f[1]=n+1;lct[0].link(1);
	m=read();
	while(m--){
		int op=read(),x=read();
		if(op) lct[col[x]].cut(x),lct[col[x]^=1].link(x);
		else{
			int v=lct[col[x]].findroot(x);
			cout<<lct[col[x]].siz[lct[col[x]].rs[v]]<<"\n";
		}
	}
	return 0;
}