标签：Ball cntb cnta int 题解 ll 舞伴 MAXN CF1475C

题意

有 \(a\) 个男孩，\(b\) 个女孩，组成了 \(k\) 对舞伴，问选出这 \(k\) 对舞伴中的两对且两对舞伴中的 \(4\) 个人互不相同的方法数。

分析

假设第 \(i\) 对舞伴由男孩 \(a[i]\) 和女孩 \(b[i]\) 组成，用 \(cnta[a[i]]\) 和 \(cntb[b[i]]\) 分别记录 \(a[i]\) 号男生出现的次数和 \(b[i]\) 号女生出现的次数。

根据上图，不难发现，与第 \(i\) 对舞伴能组成符合要求的两对舞伴的个数就是总舞伴数减去和 \(a[i]\) 重复的、和 \(b[i]\) 重复的，再加上 \(a[i]\) , \(b[i]\) 都重复的一对 \((\) 因为重复减了 \()\)，即 \(k-cnta[a[i]]-cntb[b[i]]+1\)，最后因为每对舞伴都重复计算了一遍，所以答案要除以 \(2\)。

代码

#include <bits/stdc++.h>
#define mem(a,b) memset(a,b,sizeof(a))
using namespace std;
typedef long long ll;
const int inf=1e9+7;
const int MAXN=2e5+7;
ll T,n,m,k,a[MAXN],b[MAXN],cnta[MAXN],cntb[MAXN],ans;
int main()
{
	//freopen(".in","r",stdin);
	//freopen(".out","w",stdout);
	scanf("%lld",&T);
	while(T--)
	{
		mem(cnta,0);
		mem(cntb,0);
		ans=0;
		scanf("%lld%lld%lld",&n,&m,&k);
		ll i;
		for(i=0;i<k;i++)
		{
			scanf("%lld",&a[i]);
			cnta[a[i]]++; //cnta[i]表示i号男生在舞伴中出现的次数
		}
		for(i=0;i<k;i++)
		{
			scanf("%lld",&b[i]);
			cntb[b[i]]++; //cntb[i]表示i号女生在舞伴中出现的次数
		}
		for(i=0;i<k;i++)
		{
			ans+=k-cnta[a[i]]-cntb[b[i]]+1; //计算和每一对舞伴能组成2对舞伴的舞伴数
		}
		ans/=2; //因为有重复
		printf("%lld\n",ans);
	}
	return 0;
}

标签：Ball,cntb,cnta,int,题解,ll,舞伴,MAXN,CF1475C
来源： https://www.cnblogs.com/lxy-2022/p/CF1475C-Solution.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。