C. Ball in Berland
题意:一个班级有a个男生和b个女生,且有k种男女组合(同一个男生或者女生可以出现在多个组合当中)。每个班中必须有两对组合出席舞会,且这两个组合中男生女生各不相同。让你求出一共有多少种可能的组合方式。
题解:k - 此组男生可配对的女生数-此组女生可以配对的的男生数+交集(即1),求出可以和此组一起出席的组合数,即k-P(A U B); P(A U B)=P(A)+P(B)-P(AB);统计所有的组合,因为计算过程会重复计算一次,所以要再除于2
代码:
#include<algorithm>
#include<iostream>
#include<cstring>
#include<string>
#include<cstdio>
#include<cstdio>
using namespace std;
#define MAXN 200010
int a[MAXN],b[MAXN];
int girl[MAXN],boy[MAXN];
int main(){
int t;
int n,m,k;
cin>>t;
while(t--){
cin>>n>>m>>k;
memset(girl,0,sizeof(girl));
memset(boy,0,sizeof(boy));
for(int i=1;i<=k;i++){
cin>>a[i];
boy[a[i]]++;
}
for(int i=1;i<=k;i++){
cin>>b[i];
girl[b[i]]++;
}
long long ans=0;
for(int i=1;i<=k;i++){
ans+=k-boy[a[i]]-girl[b[i]]+1;
}
cout<<ans/2<<endl;
}
}