1475C. Ball in Berland(容斥原理)

最新推荐文章于 2021-08-21 21:51:13 发布

issue是fw

最新推荐文章于 2021-08-21 21:51:13 发布

阅读量800

点赞数 2

分类专栏： div题解

本文链接：https://blog.csdn.net/jziwjxjd/article/details/113157043

版权

div题解专栏收录该内容

186 篇文章 8 订阅

订阅专栏

传送门

因为只选择两组男女关系

枚举当前选择第 $i$ 组,然后去 $[i + 1, n]$ 选择一组男女关系

如果不考虑冲突关系,答案直接加上 $n - i$

然而后面 $n - i$ 个可能和这个男生有冲突,需要减掉

后面 $n - i$ 个可能和这个女生有冲突,需要减掉

这样写可能多减了,因为有的组男女同时和第 $i$ 组冲突

所以需要加上后面 $n - i$ 组中男女关系和第 $i$ 组相同的对数

就是一个容斥原理, $P (a \cup b) = P (a) + P (b) - P (a b)$

男女关系是一个二元组,用 $m a p$ 来存

#include <bits/stdc++.h>
using namespace std;
#define int long long
const int maxn = 3e5+10;
int n1,n2,k,l[maxn],r[maxn],L[maxn],R[maxn];
typedef pair<int,int>p;
map<p,int>mp;
signed main()
{
	int t; cin >> t;
	while( t-- )
	{
		long long ans=0;
		cin >> n1 >> n2 >> k;
		for(int i=1;i<=k;i++)
			scanf("%d",&l[i]);
		for(int i=1;i<=k;i++)
			scanf("%d",&r[i]);
		for(int i=1;i<=k;i++)	mp[p(l[i],r[i])]++,L[l[i]]++,R[r[i]]++;
		for(int i=1;i<=k;i++)
		{
			mp[p(l[i],r[i])]--; L[l[i]]--; R[r[i]]--;
			if( i==k )	break;
			int pre = k-i;
			ans += pre-L[l[i]]-R[r[i]]+mp[p(l[i],r[i])];
		//	cout << pre-L[l[i]]-R[r[i]]+mp[p(l[i],r[i])] << endl;
		}
		cout << ans << endl;
		mp.clear();
	}
}