bzoj 4530 [Bjoi2014]大融合——LCT维护子树信息

题目：https://www.lydsy.com/JudgeOnline/problem.php?id=4530

LCT维护子树 siz 。设 sm[ ] 表示轻儿子的 siz 和+1(1是自己的siz)，siz[ ] 表示 splay 里 ( 两个儿子的 siz[ ] ) + sm[ cr ] 。在 access 里随便维护一下就好了。

一开始写的 siz[ ]  是 splay 里右儿子的 siz[ ] + sm[ cr ] ，但打 rev[ ]  的时候难以维护，所以弃了。

注意要先让一个点作为 splay 的根，再给它连右儿子，才能比较好地维护好；所以 link( ) 的时候注意 access( y ) ！

#include<cstdio>
#include<cstring>
#include<algorithm>
#define ll long long
using namespace std;
const int N=1e5+;
int n,Q,fa[N],c[N][],siz[N],sm[N];
int sta[N],top;bool rev[N];
int rdn()
{
  int ret=;bool fx=;char ch=getchar();
  while(ch>''||ch<''){if(ch=='-')fx=;ch=getchar();}
  while(ch>=''&&ch<='')ret=ret*+ch-'',ch=getchar();
  return fx?ret:-ret;
}
bool isroot(int x){return c[fa[x]][]!=x&&c[fa[x]][]!=x;}
void pshp(int x){siz[x]=sm[x]+siz[c[x][]]+siz[c[x][]];}
void Rev(int x)
{
  if(!rev[x])return;rev[x]=;
  rev[c[x][]]^=;rev[c[x][]]^=;
  swap(c[x][],c[x][]);
}
void rotate(int x)
{
  int y=fa[x],z=fa[y],d=(x==c[y][]);
  if(!isroot(y))c[z][y==c[z][]]=x;
  fa[x]=z;
  fa[y]=x;fa[c[x][!d]]=y;
  c[y][d]=c[x][!d];c[x][!d]=y;
  pshp(y);pshp(x);
}
void splay(int x)
{
  sta[top=]=x;
  for(int k=x;!isroot(k);k=fa[k])sta[++top]=fa[k];
  for(int i=top;i;i--)Rev(sta[i]);
  int y,z;
  while(!isroot(x))
    {
      y=fa[x];z=fa[y];
      if(!isroot(y))
    ((x==c[y][])^(y==c[z][]))?rotate(x):rotate(y);
      rotate(x);
    }
}
void access(int x)
{
  int t=;
  while(x)
    {
      splay(x);
      sm[x]+=siz[c[x][]];
      sm[x]-=siz[t];
      c[x][]=t;
      pshp(x);/////
      t=x;x=fa[x];
    }
}
void makeroot(int x)
{
  access(x);splay(x);rev[x]^=;
}
void link(int x,int y)
{
  makeroot(x);access(y);splay(y);//access()!!!
  fa[x]=y;sm[y]+=siz[x];siz[y]+=siz[x];
}
ll query(int x,int y)
{
  makeroot(x);access(y);splay(y);
  return (ll)siz[x]*sm[y];
}
int main()
{
  n=rdn();Q=rdn();
  for(int i=;i<=n;i++)siz[i]=,sm[i]=;
  int x,y;char ch[];
  while(Q--)
    {
      scanf("%s",ch);x=rdn();y=rdn();
      if(ch[]=='A')link(x,y);
      else printf("%lld\n",query(x,y));
    }
  return ;
}