Cheerleaders UVA - 11806

题目连接

Cheerleaders UVA - 11806

题意

输入n,m,k，代表有一个n*m的棋盘，有k个棋子，有多少种摆放方式？要求棋盘第一行，最后一行，第一列，最后一列都必须有棋子。

思路

从正面不太好入手，所以我们可以用容斥原理从反面考虑，设A1为第一行没有棋子的方法数，A2为第二行没有棋子的方法数，A3为第一列没有棋子的方法数，A4为最后一列没有棋子的方法数。那么最后的答案是
$ans=|S|-\sum |A_{i}|+\sum |A_{i}\cap A_{j}|-\sum |A_{i}\cap A_{j}\cap A_{k}|......$

#include<iostream>
#include<cstring>
using namespace std;
const int maxn=510;
const int mod=1000007;
int c[maxn][maxn];
void comb()
{//计算组合数，小数字直接递推即可
    int i,j;
    for(i=0;i<=500;i++)
    {
        c[i][0]=1;
        c[i][i]=1;
    }
    for(i=1;i<=500;i++)
        for(j=1;j<i;j++)
            c[i][j]=(c[i-1][j-1]+c[i-1][j])%mod;
}
int main()
{
    int T,n,m,i,j,k;
    int cas=1;
    comb();
    cin>>T;
    while(T--)
    {
        cin>>n>>m>>k;
        int sum=c[n*m][k];//没有任何限制时的结果
        for(i=1;i<(1<<4);i++)
        {//枚举
            int x=n,y=m;
            int cnt=0;//记录集合的个数，奇减偶加
            if(i&(1<<0))
                x--,cnt++;//去掉第一行
            if(i&(1<<1))
                x--,cnt++;//去掉最后一行
            if(i&(1<<2))
                y--,cnt++;//去掉第一列
            if(i&(1<<3))
                y--,cnt++;//去掉最后一列
            if(cnt&1)
                sum=((sum-c[x*y][k])%mod+mod)%mod;
            else
                sum=(sum+c[x*y][k])%mod;
        }
        cout<<"Case "<<cas++<<": "<<sum<<endl;
    }
    return 0;
}