【并查集+组合数】2019牛客暑期多校训练营(第九场) E All men are brothers

本文深入解析了一道算法竞赛题目,探讨了如何通过分析人际关系的传递性来计算互不相识组合的数量,采用动态规划和并查集的方法,详细阐述了代码实现过程。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

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


题意:给你n个人,m轮,每轮会让其中的两个人认识(人之间具有传递性),问每次挑出4个互相不认识的有多少种

思路:首先很容易确定是第一轮是C$\dbinom{4}{n}$,,(注意精度),其次我们考虑每一轮的减少量.

这时候我们考虑每次合并会带来什么影响:

假设有集合X,Y,Z,我们合并X,Y集合

那么有如下影响

合并前

1.从X集合选1个,Z集合选3个

2.从Y集合选1个,Z集合选3个

3.从X集合选1个,Y集合选1个,Z集合选两个

4.从Z集合选4个

合并后

1.从X集合选1个,Y集合选1个,Z集合选两个

也就是只有这种情况会减少,也就是减少量

减少量也就 = X,Y集合的大小相乘再乘上从其他集合任选两个的方案数

从其他集合中选出2个不在一个集合内的方案数 可以先计算任选2个的方案数,再减去来自同一个集合的方案数


#include <bits/stdc++.h>
#define endl '\n'
using namespace std;
typedef long long ll;
typedef long double ld;
typedef unsigned long long ull;
//typedef __int128 bll;
const ll maxn = 1e5+100;
const ll mod = 1e9+7;
const ld pi = acos(-1.0);
const ll inf = 1e18;
const ld eps = 1e-5;
const ld e = exp(1);

ull n,m,fa[maxn],x[maxn*2],y[maxn*2];
map<ll,ll>M;

ll fi(ll x)
{
	return fa[x] == x ? x : fa[x] = fi(fa[x]);
}

ll c2(ll x)
{
	return x*(x-1)/2;
}

int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0),cout.tie(0);
    
    cin >> n >> m;
    for(ll i = 1; i <= m; i++)
    {
    	cin >> x[i] >> y[i];
	}
    
    for(ll i = 1;i <= n; i++)
    {
    	fa[i] = i;
    	M[i] = 1;
	}
    
	ll sum = 0;//每个集合选两个的方案数总和 
   	ull ans = n*(n-1)/2*(n-2)/3*(n-3)/4;
    cout << ans << endl;
    
    for(ll i = 1; i <= m; i++)
    {
    	ll t1 = fi(x[i]),t2 = fi(y[i]);
    	
    	if(t1 != t2)
    	{
    		ll t = c2(n-M[t1]-M[t2]) - (sum - c2(M[t1]) - c2(M[t2]) ); //其他集合选两个(不来自同一集合) 
			
			ans -= M[t1]*M[t2]*t;
			sum = sum - c2(M[t1]) - c2(M[t2]) + c2(M[t1]+M[t2]); 
			
			fa[t2] = t1;
			M[t1] += M[t2];
		}
    	cout << ans << endl;
	}
 
    return 0;
}

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值