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;
}