2019牛客暑期多校训练营（第九场） E.All men are brothers

E.All men are brothers

题目链接: https://ac.nowcoder.com/acm/contest/889/E

Problem Description

Amy asks Mr. B problem E. Please help Mr. B to solve the following problem.
There are $n$ people, who don’t know each other at the beginning.
There are $m$ turns. In each turn, $2$ of them will make friends with each other.
The friend relation is mutual and transitive.
If A is a friend of B, then B is also a friend of A.
For example, if A is a friend of B, B is a friend of C, then A and C are friends.
At the beginning and after each turn, please calculate the number of ways to select four people from, such that any two of these four are not friends.

Input

The first line contains two integers, $n$ and $m(n\le 100000,m\le 200000)$, which are the number of people, and the number of turns.
In the following $m$ lines, the $i$-th line contains two integers $x$ and $y(1\le x\le n,1\le y\le n,x$ ≠ $y)$ , which means the $x$-th person and the $y$-th person make friends in the $i$-th turn.
The $x$-th person and $y$-th person might make friends in several turns.

Output

Output $m+1$ lines, each line contains an integer, which is the number of quadruples.
Output at the beginning and after each turn, so there are $m+1$ lines.

Sample Input

6 6
1 2
3 4
4 5
3 5
3 6
2 4

Sample Output

15
9
4
0
0
0
0

题意

有$n$个人，$m$回合。
在最开始的时候，$n$个人都相互不是朋友。此时输出一下从这$n$个人中选$4$个，且这$4$个人都相互不是朋友的情况的数量，也就是$C_n^4(n\ge 4)$或者$0(n<4)$。
接下来每回合给定$x$和$y$$(x$ ≠ $y)$，代表$x$和$y$成为朋友。若$A$和$B$为朋友，那么$A$和$B$的朋友也为朋友，$B$和$A$的朋友也为朋友。每回合之后，再输出从这$n$个人中选$4$个，且这$4$个人都相互不是朋友的情况的数量。

思路

若$A$和$B$为朋友，那么$A$和$B$的朋友也为朋友，$B$和$A$的朋友也为朋友。这也就代表若$x$和$y$$(x$ ≠ $y)$ 在一个集合中就代表$x$和$y$ $(x$ ≠ $y)$为朋友，这可以用并查集来判断、连接$x$和$y$。
接下来最困难的问题就是计算若$x$和$y$不是朋友，变为朋友之后对答案的减少量。
很容易可以想到相比于上一次合并集合的方案数，这一次合并集合后方案数的减少量只与$x$和$y$的集合有关。
与$x$和$y$的集合有关的方案数有三类：
$①$选中的四个集合中包含$x$所在集合，不包含$y$所在集合。（合并$x,y$集合后不影响，所以这个不必减少）
$②$选中的四个集合中包含$y$所在集合，不包含$x$所在集合。（合并$x,y$集合后不影响，所以这个不必减少）
$③$选中的四个集合同时包含$x,y$所在集合。（合并$x,y$集合后该情况不符合条件，所以减少量为此情况的数量）
所以减少量$d$就为$C_{cnt[fx]}^1\ast C_{cnt[fy]}^1\ast$(除去$x,y$集合中任选两个人不在同一集合的方案数)，其中$cnt[fx]$和$cnt[fy]$分别为$x,y$所在集合中的人数。
除去$x,y$集合中任选两个人不在同一集合的方案数=除去$x,y$集合的总人数中任选两个人减去在一个集合中选两个人的方案数的和。
除去$x,y$集合的总人数中任选两个人的方案数就是$C_{tt}^2=\frac{tt\ast (tt-1)}{2}=\frac{tt\ast tt-tt}{2}(其中tt=n-cnt[fx]-cnt[fy]),$在一个集合中选两个人的方案数的和则为${\sum }_{i=1(i不等于x和y)}^n\frac{cnt[fi]\ast (cnt[fi]-1)}{2}=\frac{{\sum }_{i=1(i不等于x和y)}^{n}cnt[fi]\ast cnt[fi]-tt}{2}$。
所以除去$x,y$集合中任选两个人不在同一集合的方案数=$\frac{tt\ast tt-tt}{2}-\frac{{\sum }_{i=1(i不等于x和y)}^{n}cnt[fi]\ast cnt[fi]-tt}{2}=\frac{tt\ast tt-{\sum }_{i=1(i不等于x和y)}^{n}cnt[fi]\ast cnt[fi]}{2}$。
设$sum$最开始为${\sum }_{i=1}^{n}cnt[fi{]}^2=n$，每次合并$x,y$后$sum$减去$cnt[fx{]}^2+cnt[fy{]}^2$就可以等于${\sum }_{i=1(i不等于x和y)}^{n}cnt[fi{]}^2$了，所以$\frac{tt\ast tt-{\sum }_{i=1(i不等于x和y)}^{n}cnt[fi]\ast cnt[fi]}{2}=\frac{tt\ast tt-sum}{2}$了，最后记得$sum$要加上$(cnt[fx]+cnt[fy]{)}^2$。

代码

#include <bits/stdc++.h>
#define pi acos(-1.0)
#define ll long long
#define ull unsigned long long
#define esp 1e-9
#define inf 0x3f3f3f3f
#define inff 0x3f3f3f3f3f3f3f3f
#define Pair pair<ll, ll>
#define It list<node>::iterator
   
using namespace std;

const ll N = 1e5+5;
ll n, m, x, y, ans, sum, f[N], cnt[N];

ll find(ll x){
	return f[x]==x ? x : f[x]=find(f[x]); 
}

int main(){
	ios::sync_with_stdio(false);
	while (cin>>n>>m){
		for (ll i = 0; i <= n; i++){
			f[i] = i; cnt[i] = 1;
		}
		ll tmp = n;
		if (tmp < 4){//最开始就小于4个集合
			ans = 0;
		}
		else{
			ans = 1;
			sum = n;//初始化sum为n
			ll vis[5] = {0};
			for (ll i = n; i >= n-3; i--){//计算最开始的方案数
				ll t = i;
				for (ll j = 2; j <= 4; j++){
					if (!vis[j] && t%j==0){
						vis[j] = 1;
						t /= j;
					} 
				}
				ans *= t;
			}
		}
		cout<<ans<<endl;
		while (m--){
			cin>>x>>y;
			ll fx = find(x), fy = find(y), d = 0;
			if (fx != fy){//x,y集合要合并
				ll tt = n-cnt[fx]-cnt[fy];//计算除去x,y集合的总人数tt
				sum -= (cnt[fx]*cnt[fx]+cnt[fy]*cnt[fy]);//更新sum
				d = 1LL*cnt[fx]*cnt[fy]*((tt*tt-sum)/2);//计算方案的减少量
				sum += ((cnt[fx]+cnt[fy])*(cnt[fx]+cnt[fy]));//sum加回(cnt[fx]+cnt[fy])*(cnt[fx]+cnt[fy])
				tmp--;
				f[fx] = fy;//将x合并进y所在集合
				cnt[fy] += cnt[fx];//y所在集合人数加上x所在集合的人数
			}
			ans -= d;
			cout<<ans<<endl;
		}
	}
 
	return 0;
}