P4219大融合-LCT维护子树信息

P4219-LCT维护子树信息

题目大意

给定 $n$个结点和 $q$ 次操作，每个操作为如下形式：
$Axy$ 在结点 $x$和$y$之间连接一条边。
$Qxy$ 给定一条已经存在的边 $(x,y)$，求有多少条简单路径，其中包含边 $(x,y)$ 。
保证在任意时刻，图的形态都是一棵森林。

题解

该题简单来说，也就是LCT维护子树信息
因为最终的答案就是$x$节点的子树大小$\ast y$节点子树的大小
然而一般的LCT都是维护链上信息，那么如何做到维护子树信息呢

我们用$siz1[x]$表示$x$节点总的子树和，$siz2[x]$表示$x$节点的虚儿子的子树和
然后对其进行维护，详情参见代码

代码

#include<bits/stdc++.h>
#define int long long
#define M 1000009
using namespace std;
int read(){
	int f=1,re=0;char ch;
	for(ch=getchar();!isdigit(ch)&&ch!='-';ch=getchar());
	if(ch=='-'){f=-1,ch=getchar();}
	for(;isdigit(ch);ch=getchar()) re=(re<<3)+(re<<1)+ch-'0';
	return re*f;
}
int n,rev[M],c[M][2],f[M],q[M],m,siz1[M],siz2[M];
char s[20];
void pushup(int x){siz1[x]=siz2[x]+siz1[c[x][1]]+siz1[c[x][0]]+1;}
bool root(int x){return c[f[x]][0]!=x&&c[f[x]][1]!=x;} 
void pushdown(int x){
	if(rev[x]){
		rev[c[x][0]]^=1,rev[c[x][1]]^=1;
		swap(c[x][0],c[x][1]);
		rev[x]^=1;
	}
}
void rotate(int x){
	int fa=f[x],gfa=f[fa];
	int l=c[fa][1]==x, r=l^1;
    if(!root(fa)) c[gfa][c[gfa][1]==fa]=x;
    f[x]=gfa,f[c[x][r]]=fa,f[fa]=x;
    c[fa][l]=c[x][r];c[x][r]=fa; 
    pushup(fa);
}
void splay(int x){
	int top=0;
	q[++top]=x;
	for(int i=x;!root(i);i=f[i]) q[++top]=f[i];
	while(top) pushdown(q[top--]);
	while(!root(x)){
		int fa=f[x],gfa=f[fa];
		if(!root(fa)){
			if(c[fa][0]==x^c[gfa][0]==fa) rotate(x);
			else rotate(fa);
		}rotate(x);
	}pushup(x);
}
void access(int x){
	for(int y=0;x;y=x,x=f[x])
		splay(x),siz2[x]+=siz1[c[x][1]],siz2[x]-=siz1[c[x][1]=y];
}
void makeroot(int x){access(x),splay(x),rev[x]^=1;}
void split(int x,int y){makeroot(x),access(y),splay(y);}
void link(int x,int y){split(x,y),f[x]=y,siz2[y]+=siz1[x],pushup(y);}
//void cut(int x,int y){makeroot(x),access(y),splay(y),c[y][0]=f[x]=0;}
//int findroot(int x){
//	access(x),splay(x);
//	while(c[x][0]) x=c[x][0];
//	return x;
//}
signed main(){
	n=read(),m=read();
	for(int i=1;i<=n;i++) siz1[i]=1;
	for(int i=1;i<=m;i++){
		scanf("%s",s);
		int x=read(),y=read();
		if(s[0]=='A') link(x,y);
		if(s[0]=='Q') split(x,y),printf("%lld\n",(siz2[x]+1)*(siz2[y]+1));
	}return 0;
}