luogu P4219 [BJOI2014]大融合

背景：

$LCT$维护子树信息的模板题。

题目传送门：

https://www.luogu.org/problemnew/show/P4219

题意：

动态加边的一棵树，求当前联通时这一条边的负载（任意两点的路径经过这条边的次数）。

思路：

动态加边，显然用$LCT$，然而$LCT$只 能维护重链啊。
怎么办。
我们知道轻重儿子（轻重链）是不断变化的，因此，在每一次转换轻重儿子时记录一下，更改一下即可。
用$size$记录轻儿子所连的子树大小，$Size$记录整棵子树的大小。
那么这三个函数相应变成：

void pushup(int x)
{
	tr[x].Size=tr[tr[x].son[0]].Size+tr[tr[x].son[1]].Size+tr[x].size+1;
}

void access(int x)
{
	for(int y=0;x;y=x,x=tr[x].fa)
		splay(x),tr[x].size+=tr[tr[x].son[1]].Size,tr[x].size-=tr[tr[x].son[1]=y].Size,pushup(x);
}

void link(int x,int y)
{
	makeroot(x);
	makeroot(y);
	tr[x].fa=y;
	tr[y].size+=tr[x].Size;
	pushup(y);
}

理解起来不难，因为父亲只存重儿子，因此就有：$Siz{e}_{now}=Siz{e}_{lc}(左儿子)+Siz{e}_{rc}(右儿子)+siz{e}_{now}(轻儿子所连的子树大小)+1(自己)$。

而$access$操作将$x$到根的路径变为重链，因而在$now$的右儿子变为轻儿子之前，$size$需要加上它，而变完后$size$就要减少变为重儿子的那一部分。

$link$操作是加上一条轻链，因而改变的是$size$，由定义可知要加上$Size$。

因为$del$操作之前要$split$，将其变为重链，$size$在$access$中更新，因此不受影响，不需要改。

求解操作就是一个乘法原理，没什么。

代码：

#include<cstdio>
#include<cstring>
#include<algorithm>
#define LL long long
using namespace std;
	struct node{int d,fa,size,Size,op,son[2];} tr[100010];
	int n,m;
bool isroot(int x)
{
	return tr[tr[x].fa].son[0]!=x&&tr[tr[x].fa].son[1]!=x;
}
bool isson(int x)
{
	return x==tr[tr[x].fa].son[1];
}
void change(int x)
{
	if(!x)return;
	swap(tr[x].son[0],tr[x].son[1]);
	tr[x].op^=1;
}
void pushup(int x)
{
	tr[x].Size=tr[tr[x].son[0]].Size+tr[tr[x].son[1]].Size+tr[x].size+1;
}
void pushdown(int x)
{
	if(!tr[x].op) return;
	change(tr[x].son[0]),change(tr[x].son[1]);
	tr[x].op=0;
}
void rot(int x)
{
	int w=isson(x),y=tr[x].fa,yy=tr[y].fa;
	tr[y].son[w]=tr[x].son[w^1];
	if(tr[y].son[w]) tr[tr[y].son[w]].fa=y;
	tr[x].fa=yy;
	if(!isroot(y)) tr[yy].son[isson(y)]=x;
	tr[x].son[w^1]=y;tr[y].fa=x;
	pushup(y);
}
int sta[300010];
int top;
void splay(int x)
{
	sta[top=1]=x;
	for(int i=x;!isroot(i);i=tr[i].fa)
		sta[++top]=tr[i].fa;
	while(top) pushdown(sta[top--]);
	for(int y=tr[x].fa;!isroot(x);rot(x),y=tr[x].fa)
		if(!isroot(y)) isson(x)^isson(y)?rot(x):rot(y);
	pushup(x);
}
void access(int x)
{
	for(int y=0;x;y=x,x=tr[x].fa)
		splay(x),tr[x].size+=tr[tr[x].son[1]].Size,tr[x].size-=tr[tr[x].son[1]=y].Size,pushup(x);
}
void makeroot(int x)
{
	access(x);splay(x);change(x);
}
void link(int x,int y)
{
	makeroot(x);
	makeroot(y);
	tr[x].fa=y;
	tr[y].size+=tr[x].Size;
	pushup(y);
}
void split(int x,int y)
{
	makeroot(x);access(y);splay(y);
}
void del(int x,int y)
{
	split(x,y);
	tr[x].fa=tr[y].son[0]=0;
}
LL work(int x,int y)
{
	del(x,y);
	makeroot(x);makeroot(y);
	LL ans=(LL)tr[x].Size*tr[y].Size;
	link(x,y);
	return ans;
}
int main()
{
	char s[5];
	int x,y;
	scanf("%d %d",&n,&m);
	tr[0]=(node){0,0,0,0,0,0,0};
	for(int i=1;i<=n;i++)
		tr[i]=(node){0,0,0,1,0,0,0};
	for(int i=1;i<=m;i++)
	{
		scanf("%s %d %d",s+1,&x,&y);
		if(s[1]=='A') link(x,y); else printf("%lld\n",work(x,y));
	}
}