hdu 4451 Dressing(容斥原理)

注意:本题只有注意只有“clothes x pants y” or “pants y shoes z”  这两种输入!没看到这个想了很久,一定要注意读题啊!


用容斥原理先把所有的都正常减,再把重复的加回去。

a数组记录某中间结点和左边的集合有多少条连边,b数组记录和右边的有多少连边。这样对于结点i,重复的方案数为a【i】*b【i】把这些都加回去就好了。


代码:

#include <cstdio>
#include <iostream>
#include <cstring>
using namespace std;

int a[1005],b[1005];
int main(){
    int n,m,k;
    while(~scanf("%d%d%d",&n,&m,&k)){
        if(!n&&!m&&!k) break;
        int P;
        scanf("%d",&P);
        int res=n*m*k;
        memset(a,0,sizeof(a));
        memset(b,0,sizeof(b));
        while(P--){
            char t1[10],t2[10];
            int x,y;
            scanf("%s%d%s%d",t1,&x,t2,&y);
            if(t1[0]=='c'&&t2[0]=='p'){
                a[y]++;
                res-=k;
            }
            else{
                b[x]++;
                res-=n;
            }
        }
        for(int i=1;i<=m;i++){
            res+=a[i]*b[i];
        }
        printf("%d\n",res);
    }
    return 0;
}


你可能感兴趣的:(容斥原理)