UVA - 11806 Cheerleaders【容斥】

题目传送门

题目描述：t组测试数据，每组数据输入三个数M，N，K;表示有一个大小为M*N的棋盘和k枚棋子，2 ≤ M, N ≤ 20 ， K ≤ 500. 将K枚棋子放入棋盘中，每个格子最多只能放一枚棋子，要求最外面的4条边每一条边上至少有一个棋子，若棋子恰巧在两条边的拐角处，那么这枚棋子可以覆盖这两条边。问有多少种放置方式，输出对1000007取模后的结果。

解题思路：所有可能的放置方式减去不符合要求的放置方式。

c[i][j]为组合数，表示从大小为i的棋盘中任意选取j个格子放置棋子，可用杨辉三角来求。

1. 所有可能的放置方式：k枚棋子任意放置在大小为m*n的棋盘上 A=c[m*n][k];

2. 有一条边没有被放置：B=(c[m*(n-1)][k]+c[(m-1)*n][k])*2;

3. 有两条边没有被放置：C=c[m*(n-2)][k]+c[(m-2)*n][k]+c[(m-1)*(n-1)][k]*4;

4. 有三条边没有被放置：D=(c[(m-1)*(n-2)][k]+c[(m-2)*(n-1)][k])*2;

5. 四条边都没有被放置：E=c[(m-2)*(n-2)][k];

显而易见的是这几种情况中有很多重复的地方，比如：有一条边没有被放置包括有两条边没有被放置、有三条边没有被放置、四条边都没有被放置，而有两条边没有被放置又包括有三条边没有被放置、四条边都没有被放置，同理有三条边没有被放置包括四条边都没有被放置。这就要用到容斥原理，最终结果ans=A-B+C-D+E。

AC代码：

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <cstring>
#include <string>
#include <cmath>
#include <queue>
#include <stack>
#include <vector>
#include <map>
#include <set>
using namespace std;
#define io ios::sync_with_stdio(0),cin.tie(0)
#define ms(arr) memset(arr,0,sizeof(arr))
#define inf 0x3f3f3f
typedef long long ll;
const int mod=1e6+7;
const int maxn=1e5+7;
int t,n,m,k;
int c[500][500];
void getC()//杨辉三角求组合数
{
    for(int i=0; i<500; 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()
{
    io;
    getC();
    cin>>t;
    for(int o=1; o<=t; o++)
    {
        cin>>n>>m>>k;
        int A=(c[n*m][k])%mod;//所有可能的放置方式
        int B=(2*(c[(n-1)*m][k]+c[n*(m-1)][k]))%mod;//有一条边没有被放置
        int C=(c[(n-2)*m][k]+4*c[(n-1)*(m-1)][k]+c[n*(m-2)][k])%mod;//有两条边没有被放置
        int D=(2*(c[(n-2)*(m-1)][k]+c[(n-1)*(m-2)][k]))%mod;//有三条边没有被放置
        int E=(c[(n-2)*(m-2)][k])%mod;//四条边都没有被放置
        int ans=(A-B+C-D+E+mod)%mod;
        cout<<"Case "<<o<<": "<<ans<<endl;
    }
    return 0;
}