题目大意:给出衣服、裤子和鞋子的数量n,m,k,再给出不能搭配的对应关系,问有多少种搭配方式。
解题思路:这是一个容斥原理,这里对裤子进行枚举。接收不能搭配关系时,存储不能搭配衣服的鞋子和不能搭配鞋子的衣服。然后枚举每一条裤子,看能够搭配该裤子的衣服和鞋子各有多少种。详见code。
题目来源:http://acm.hdu.edu.cn/showproblem.php?pid=4451
code:
#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
#define ll __int64
const int MAXN = 1000+10;
int n,m,k,pp,num1,num2;
char str1[MAXN],str2[MAXN];
int c[MAXN],s[MAXN];
ll ans;
int main(){
//freopen("input.txt","r",stdin);
while(~scanf("%d%d%d",&n,&m,&k) && n+m+k){
getchar();
ans=0;
memset(c,0,sizeof(c));
memset(s,0,sizeof(s));
scanf("%d",&pp);
getchar();
for(int i=0;i<pp;++i){
scanf("%s %d %s %d",str1,&num1,str2,&num2);
if(str1[0]=='c') c[num2]++; //衣服不能搭配的鞋子数
if(str2[0]=='s') s[num1]++; //鞋子不能搭配的衣服数
}
for(int i=1;i<=m;++i) //枚举裤子,有多少能够搭配的衣服和鞋子数
ans+=(n-c[i])*(k-s[i]);
printf("%I64d\n",ans);
}
return 0;
}