HDU 4451 容斥原理

题目大意：

n件衣服，m条裤子，k双鞋子进行搭配

妈妈指明了哪些衣服和裤子不能搭配，哪些裤子和鞋子不能搭配，问最后有几种搭配方法

 

先假设都能搭配 n*m*k

每次遇到衣服和裤子不能搭的，就要减一次k，同时记录这个衣服和裤子出现的次数，避免以后重复减去的可以加回来

裤子和鞋子也是同样道理

 

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
const int N=1005;
int cl[N][2] , pa[N][2] , sh[N][2];
int n , m , k , p , a , b;
char s1[20], s2[20];

int main()
{
   // freopen("a.in" , "r" , stdin);
    while(scanf("%d%d%d" , &n , &m , &k) ,n||m||k)
    {
        memset(cl , 0 ,sizeof(cl));
        memset(pa , 0 ,sizeof(pa));
        memset(sh , 0 ,sizeof(sh));

        int ans = n*m*k;
        scanf("%d" , &p);
        for(int i=0 ; i<p ; i++)
        {
            scanf("%s%d%s%d" , s1 , &a , s2 , &b);
            if(s1[0] == 'c' && s2[0] == 'p'){
                ans -= k;
                ans += cl[a][1];
                ans += pa[b][1];
                cl[a][0]++;
                pa[b][0]++;
            }
            else if(s1[0] == 'p' && s2[0] == 's'){
                ans -= n;
                ans += pa[a][0];
                ans += sh[b][0];
                pa[a][1]++;
                sh[b][1]++;
            }
        }
        printf("%d\n" , ans);
    }
    return 0;
}