UVA 11806 - Cheerleaders(容斥原理)

题意：  在m行n列的矩形网格中放置k个相同的石子，要求第一行、最后一行、第一列、最后一列都必须有石子，问一共有多少种放置的方法。

分析： 由容斥原理，设第一行没有石子的方法数为A，最后一行没有石子的方法数为B，第一列没有石子的方法数为C，最后一列没有石子的方法数为D，不加任何限制而放置k个石子的总方法数为S。那么答案应该为S - (A U B U C U D)。用二进制的方式表示集合A、B、C、D的组合方式。0001表示在A中、0010表示在B中、0100表示在C中、1000表示在D中、0101表示在A和C中、0111表示在A、B和C中、…

#include<bits/stdc++.h>

using namespace std;

int c[500][500];

const int mod=1e6+7;

void init(){
    c[0][0]=1;
    for(int i=1;i<=410;i++){
        c[i][0]=c[i][i]=1;
        for(int j=1;j<i;j++){
            c[i][j]=(c[i-1][j]+c[i-1][j-1])%mod;
        }
    }
}

int main(){
    init();
    int T;
    scanf("%d",&T);
    for(int cs=1;cs<=T;cs++){
        int m,n,k;
        scanf("%d%d%d",&m,&n,&k);
        int ans=c[m*n][k];
        for(int i=1;i<16;i++){
            int cnt=0,row=m,col=n;
            if(i&1) row--,cnt^=1;
            if(i&2) row--,cnt^=1;
            if(i&4) col--,cnt^=1;
            if(i&8) col--,cnt^=1;
            if(cnt&1) ans=(ans-c[row*col][k]+mod)%mod;
            else ans=(ans+c[row*col][k]+mod)%mod;
        }
        printf("Case %d: %d\n",cs,ans);
    }
    return 0;
}