bzoj4530/lg4291
需要维护子树个数..
在lct中维护两个值:size[x](x的所有子树个数)、isize[x](与x通过虚边链接的子树个数)
考虑LCT基本操作中哪些需要变
update肯定会变
inline void update(int x){
size[x]=size[ch[x][0]]+size[ch[x][1]]+isz[x]+1; //要加上isz[x],表示连到x的虚树的个数
}
link操作中,把x、y之间建一条虚边,虚树个数也会改变
inline void link(int x,int y){
makeroot(x);makeroot(y); //注意这里要把y也换到跟,不然y上面的信息就不对了
fa[x]=y;isz[y]+=size[x];update(y); //x向上连一条虚边连到y,y的虚树多了x,所以要先处理y的虚树个数,在处理y的所有数个数(update中会用到isz啊...)
}
access是把x通到根,显然会改变虚实关系,也是要改哦
inline void access(int x){
int y=0;
while(x){
splay(x);isz[x]+=size[ch[x][1]]; //要把ch[x][1]变为虚边了~
ch[x][1]=y;isz[x]-=size[y];update(x); //y变为实边了呢~
y=x;x=fa[x];
}
}
cut,已经没有变了…虚边都不是了..update维护就好啦,所以不用变
splay,显然是在实树的时候才会进行的,因此不会变
makeroot,在access时已经变过来啊,那就不变了吧
所以最终,代码如下~
#include <cstdio>
#define N 100010
int n,m,ch[N][2],size[N],isz[N],fa[N],rev[N];
inline bool isroot(int x){return x!=ch[fa[x]][1] && x!=ch[fa[x]][0];}
inline void pushdown(int x){
if(!rev[x]) return;
rev[ch[x][0]]^=1;rev[ch[x][1]]^=1;rev[x]=0;
}
void push(int x){
if(!isroot(x)) push(fa[x]);
pushdown(x);
}
inline void update(int x){size[x]=size[ch[x][0]]+size[ch[x][1]]+isz[x]+1;}
inline void rotate(int x){
int y=fa[x],z=fa[y],t=ch[y][0]==x;
if(!isroot(y)) ch[z][ch[z][1]==y]=x;
fa[y]=x;fa[x]=z;fa[ch[x][t]]=y;
ch[y][t^1]=ch[x][t];ch[x][t]=y;
update(y);update(x);
}
inline void splay(int x){
push(x);
while(!isroot(x)){
int y=fa[x];
if(isroot(y)){rotate(x);return;}
if(ch[y][0]==x^ch[fa[y]][0]==y) rotate(x);
else rotate(y);rotate(x);
}
}
inline void access(int x){
int y=0;
while(x){
splay(x);isz[x]+=size[ch[x][1]];
ch[x][1]=y;isz[x]-=size[y];update(x);
y=x;x=fa[x];
}
}
inline void makeroot(int x){
access(x);splay(x);rev[x]^=1;
}
inline void link(int x,int y){
makeroot(x);
makeroot(y);
fa[x]=y;isz[y]+=size[x];update(y);
}
inline void query(int x,int y){
makeroot(x);access(y);splay(y);
printf("%d\n",size[x]*(size[y]-size[x]));
}
int main(){
scanf("%d%d",&n,&m);
// for(int i=1;i<=n;i++) size[i]=1;
for(int i=1;i<=m;i++){
char op[5];int x,y;scanf("%s%d%d",op,&x,&y);
if(op[0]=='A') link(x,y);
else query(x,y);
}
return 0;
}