容斥

最新推荐文章于 2022-04-12 22:05:46 发布

蒟蒻大法好

最新推荐文章于 2022-04-12 22:05:46 发布

阅读量221

点赞数

文章标签：容斥

本文链接：https://blog.csdn.net/qq_41650771/article/details/97922309

版权

题目链接

题目大意,给一个n*m的格子k个石子,要求第一行最后一行,第一列最后一列必须有石子,问有多少种方案
思路：
假设有四个集合 A,B,C,D分别表示不在第一行,不在最后一行,不在第一列不在最后一列的方案,总集为S,那么我们要求的答案肯定就是属于S但是不属于 ABCD 中的任何一个。
ABCD总共有16六种情况我们可以枚举4位二进制分别代表又或者没有,根据容斥原理,假如0001代表第一行没有0010代表最后一行没有那么0011代表两行都没有,这种情况我们会多减一次，所以我们要把这种情况的方案数重新加回来。
代码：

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int mod = 1000007;
ll C[1004][1005];
void init()
{
    for(int i = 0; i <= 1000; i ++) C[i][0] = 1;//杨辉三角
    for(int i = 0; i <= 1000 ; i ++){
        C[i][i] = 1;
        for(int j = 1; j <= i; j ++){
            C[i][j] = (C[i -1][j - 1] + C[i - 1][j])%mod;
        }
    }
}
int main()
{
    init();
    int t;
    cin >>t;
    int num = 0;
    while(t --){
        cout <<"Case "<< ++num <<": ";
        ll sum = 0;
        int n,m,k;
        cin >> n >> m >>k;
        for(int i = 0; i < 16; i ++){
            int r = n,c = m, b = 0;
            if(i & 1) r -- ,b ++;//四种情况
            if(i & 2) r -- ,b ++;
            if(i & 4) c -- , b ++;
            if(i & 8) c --, b ++;
            if(b&1) sum = (sum + mod - C[c*r][k]) % mod;
            else sum = (sum + C[c*r][k]) % mod;

        }
        cout << sum << endl;
    }
    return 0;
}