1598D Training Session Educational Codeforces Round 115 (Rated for Div. 2) 组合数学+暴力

题意
有n个题目组成的题目集，每个题目有一个topic和一个diffcult，同一个topic中的题目不会有相同的diffcult，要求取3个题目出来对应的topic都不同或者difficulty都不同有多少种取法。
思路
题目要求选3个题目出来对应的topic都不同或者对应的difficulty都不同，我们反向考虑，选3个题目出来，题目对应的topic和difficulty都有相同，然后用方案总数-topic和diffculty都有相同的方案数。因为同一个topic中的difficulty都不同，所以3个问题中只能有两个题目topic相同且这两个题目diffculty肯定不同，difficulty也只有两个题目相同。
3个问题中两个题目topic相同，这两个题目对应的difficulty肯定不同，那么就要从其他topic中取一个和这两个其中一个difficulty相同的问题，这样子考虑还是不够清晰，我们取一个题目的difficulty和另一个题目的difficulty相同，再从随便取一个和这个题目topic相同的题目，这样子就清晰了。下面看伪代码：

	long long ans;//方案总数
	long long ans2;//topic和diffculty都有相同的方案数
	long long rans;//符合题意的方案数，rans=ans-ans2
	long long num[200005];//topic为i的数量
	for(int i 1~n)//枚举每个题目
	{
		for(int j 1~n)//寻找不是第i个题目但是和第i个题目diffculty相同的题目
		{
			if(i!=j&&diffculty[i]==diffculty[j])
			{
				ans2+=num[topic[i]]-1；
			}
		}
	}
	rans=ans-ans2

上面伪代码转化为代码写出来交上去肯定是要超时的，但是我们可以看到第二维循环我们要找的是和第i个题目diffculty相同的题目，我们可以用一个数组取存储mp[v]存储难度为v的题目个数，如果一个题目i的diffculty为v，那么不是题目i但是和题目i的diffculty相同的题目个数为mp[v]-1.再看伪代码：

long long ans;//方案总数
	long long ans2;//topic和diffculty都有相同的方案数
	long long rans;//符合题意的方案数，rans=ans-ans2
	long long num[200005];//topic为i的数量
	long long mp[200005];
	for(int i 1~n)//枚举每个题目
	{
		ans2+=(num[topic[i]]-1)*(mp[diffculty[i]-1);
	}
	rans=ans-ans2

代码

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
int T;
ll n;//题目数，要开longlong
ll mp[200005];//mp[i] diffculty为i的题目数量
ll num[200005];//num[i] topic为i的题目数量
void init()//初始化
{
	for(int i=1;i<=n;i++)
	num[i]=0,mp[i]=0;
}
struct node
{
	int a;//topic
	int b;//diffculty
}que[200005];
int main()
{ 
	scanf("%d",&T);
	while(T--)
	{
		scanf("%lld",&n);
		init();
		ll ans=n*(n-1)*(n-2)/6;//n个题目中任取3个方案总数
		ll ans2=0;//不符合题意的方案数
		for(int i=1;i<=n;i++)//输入题目数据并初始化num和mp
		{
			scanf("%d%d",&que[i].a,&que[i].b);
			num[que[i].a]++;
			mp[que[i].b]++;
		}
		for(int i=1;i<=n;i++)//计算不符合题意的方案总数
		{
			int a=que[i].a;
			int b=que[i].b;
			ans2+=(num[a]-1)*(mp[b]-1);
		}
		printf("%lld\n",ans-ans2);
	}

return 0;
}