【bzoj4530】大融合（LCT的子树维护）

LCT维护子树并没有想象中的那么难，在这里只是复习下。

（其他的维护子树的题目，可见：“共价大爷游长沙”）

只要记录下虚边连接的信息就好了。

 #include<bits/stdc++.h>
 #define N 100010
 typedef long long ll;
 using namespace std;
 int n,q,ans,tot,num;
 struct Link_Cut_Tree{
     int c[N][],fa[N],size1[N],size2[N],q[N],top,rev[N];
     inline void pushup(int x){
         int l=c[x][],r=c[x][];
         size1[x]=size1[l]+size1[r]+size2[x];
     }
     inline bool isroot(int x){return c[fa[x]][]!=x&&c[fa[x]][]!=x;}
     inline void pushdown(int x){
         int l=c[x][],r=c[x][];
         if(rev[x]){
             rev[l]^=;rev[r]^=;rev[x]^=;
             swap(c[x][],c[x][]);
         }
     }
     inline void rotate(int x){
         int y=fa[x],z=fa[y],l,r;
         if(c[y][]==x)l=;else l=;r=l^;
         if(!isroot(y)){if(c[z][]==y)c[z][]=x;else c[z][]=x;}
         fa[x]=z;fa[y]=x;fa[c[x][r]]=y;
         c[y][l]=c[x][r];c[x][r]=y;
         pushup(y);pushup(x);
     }
     inline void splay(int x){
         top=;q[top]=x;
         for(int i=x;!isroot(i);i=fa[i])q[++top]=fa[i];
         for(int i=top;i;i--)pushdown(q[i]);
         while(!isroot(x)){
             int y=fa[x],z=fa[y];
             if(!isroot(y)){
                 if((c[z][]==y)^(c[y][]==x))rotate(x);
                 else rotate(y);
             }rotate(x);
         }
         pushup(x);
     }
     void access(int x){
         for(int t=;x;t=x,x=fa[x]){
             splay(x);size2[x]+=size1[c[x][]];
             c[x][]=t;size2[x]-=size1[t];pushup(x);
         }
     }
     void makeroot(int x){access(x);splay(x);rev[x]^=;}
     int find(int x){access(x);splay(x);while(c[x][])x=c[x][];return x;}
     void split(int x,int y){makeroot(x);access(y);splay(y);}
     void link(int x,int y){
         makeroot(x);makeroot(y);fa[x]=y;size2[y]+=size1[x];
         pushup(y);
     }
     void cut(int x,int y){split(x,y);if(c[y][]==x)c[y][]=,fa[x]=;}
     void init(int n){
         for(int i=;i<=n;i++)size1[i]=size2[i]=;
     }
 }T;
 inline int read(){
     int f=,x=;char ch;
     do{ch=getchar();if(ch=='-')f=-;}while(ch<''||ch>'');
     do{x=x*+ch-'';ch=getchar();}while(ch>=''&&ch<='');
     return f*x;
 }
 int main(){
     n=read();q=read();
     T.init(n);char s[];
     while(q--){
         scanf("%s",s);
         int x=read(),y=read();
         if(s[]=='A')T.link(x,y);
         else{
             T.makeroot(x);T.access(y);T.splay(x);
             printf("%lld\n",(ll)((T.size2[y])*(T.size1[x]-T.size2[y])));
         }
     }
     return ;
 }