题目描述:
有N个人M个操作,每次操作让两个人互相认识,认识关系可以互相传递,求每次操作完毕后,选4个互相不认识的人的方案数
题目分析:
并查集维护联通块内的人数
刚开始答案一定是 C(N,4)
我们定义一个数量Pre为目前选两个互相不认识的人的方案数 初始值为C(N,2)
每次合并两个联通块,如果两个人本来就在一个联通块里是不需要改变答案的。
如果没有在一个联通块内,我们考虑如何通过上次的答案求出这次的。
很明显,我们设A为联通块X的人数,B为联通块B的人数
对于答案内,我们要去掉的就是AB互相不认识的两个人的数量(去除AB后
那么也就是
A
∗
B
∗
(
P
r
e
−
A
∗
(
n
−
A
)
−
B
∗
(
n
−
B
)
)
A*B*(Pre-A*(n-A)-B*(n-B))
A∗B∗(Pre−A∗(n−A)−B∗(n−B))了,
当然不要忘记更新Pre的值
另外long long不够用,可以开long double,不知道是牛客不支持__int128还是我写的有问题,反正我用__int128WA了
代码:
#include <cstdio>
#include <iostream>
#include <algorithm>
#define ld long double
const int maxm=110000;
ld n,siz[maxm];
int m;
int fa[maxm];
inline void init(){for(int i=1;i<=n;i++) fa[i]=i,siz[i]=1;}
int find(int x){return x==fa[x]?x:fa[x]=find(fa[x]);}
signed main()
{
scanf("%Lf%d",&n,&m);
ld ans=n*(n-1)*(n-2)*(n-3)/24LL;
ld pre=n*(n-1)/2;
printf("%.0llf\n",ans);
init();
for(int i=1;i<=m;i++)
{
int x,y;
scanf("%d%d",&x,&y);
if(find(x)==find(y))
{
printf("%.0llf\n",ans);
continue;
}
ld a=siz[find(x)],b=siz[find(y)];
ans-=a*b*(pre-a*(n-a)-b*(n-b)+a*b);
pre-=a*b;
printf("%.0llf\n",ans);
siz[find(y)]+=siz[find(x)];
fa[find(x)]=find(y);
}
return 0;
}