UVA11806 Cheerleaders（容斥）

题意：

给定n,m,k，
表示有一个n*m的矩阵，
现在你要在矩阵中选出k个不同的格子放k个人，

要求矩阵中四条边上都至少有一个人
(4个角的格子可以算作在两个边上)

答案对1e6+7取模

数据范围：n,m<=20，k<=500

解法：

答案=C(n*m,k)-至少一条边没人+至少两条边没人-至少三条边没人+至少四条边没人

用4位二进制数表示4条边放不放,二进制枚举一下就行了
(根据二进制1的位数,奇减偶加)

code：

#include<bits/stdc++.h>
using namespace std;
#define int long long
const int maxm=505;
const int mod=1e6+7;
int n,m,k;
int c[maxm][maxm];
void init(){
    c[0][0]=1;
    for(int i=1;i<maxm;i++){
        c[i][0]=1;
        for(int j=1;j<=i;j++){
            c[i][j]=c[i-1][j-1]+c[i-1][j];
            c[i][j]%=mod;
        }
    }
}
int cal(int x){
    int ans=0;
    while(x){
        ans++;
        x&=(x-1);
    }
    return ans;
}
signed main(){
    ios::sync_with_stdio(0);
    init();
    int T;cin>>T;
    int cas=1;
    while(T--){
        cin>>n>>m>>k;
        cout<<"Case "<<cas++<<": ";
        if(k>n*m){
            cout<<0<<endl;
            continue;
        }
        int ans=c[n*m][k];
        for(int i=1;i<(1<<4);i++){
            int x=n,y=m;
            for(int j=0;j<4;j++){
                if(i>>j&1){//=1表示不选
                    if(j==0||j==1)x--;//行不选
                    else y--;//列不选
                }
            }
            int temp=c[x*y][k];
            if(cal(i)&1)ans-=temp;
            else ans+=temp;
        }
        cout<<(ans%mod+mod)%mod<<endl;
    }
    return 0;
}

---