HDU 6314 Matrix(容斥原理)

本文介绍了一种计算特定条件下的01矩阵数量的方法。通过选取指定行和列全为1的矩阵,利用组合数学和容斥原理计算方案数,并通过预处理组合数和2的幂次来优化算法。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

Description

问至少有AAB列全是11n×m0101矩阵个数

Input

多组用例,每组用例输入四个整数n,m,A,B(1n,m,A,B3000)n,m,A,B(1≤n,m,A,B≤3000)

Output

输出方案数,结果模998244353998244353

Sample Input

3 4 1 2

Sample Output

169

Solution

先考虑行,选出ii行为黑、至少b列为黑的方案数为CinCni⋅所选ii行为黑、至少b列为黑的方案数,但在选j<ij<i行为黑时会重复计算,考虑容斥,假设每部分实际应该计算的次数为fifi次,那么有fi=1k=ai1fkCkifi=1−∑k=ai−1fkCik,同理可以求出选出jj列为黑实际被计算的次数gj=1k=bj1gkCjk,而所选iij列为黑的方案数即为2(ni)(mj)2(n−i)(m−j),进而答案即为ans=i=anj=bmfigjCinCjm2(ni)(mj)ans=∑i=an∑j=bmfigjCniCmj2(n−i)(m−j),预处理组合数和22的幂次即可,方案数O(n2+m2+nm)

Code

#include<cstdio>
using namespace std;
typedef long long ll;
#define maxn 3005
#define mod 998244353
int mul(int x,int y)
{
    ll z=1ll*x*y;
    return z-z/mod*mod;
}
int add(int x,int y)
{
    x+=y;
    if(x>=mod)x-=mod;
    return x;
}
int C[maxn][maxn],f2[maxn*maxn],f[maxn],g[maxn];
void init(int n=3000)
{
    f2[0]=1;
    for(int i=1;i<=n*n;i++)f2[i]=add(f2[i-1],f2[i-1]);
    C[0][0]=1;
    for(int i=1;i<=n;i++)
    {
        C[i][0]=C[i][i]=1;
        for(int j=1;j<i;j++)C[i][j]=add(C[i-1][j-1],C[i-1][j]);
    } 
}
int main()
{
    init();
    int n,m,a,b;
    while(~scanf("%d%d%d%d",&n,&m,&a,&b))
    {
        for(int i=a;i<=n;i++)
        {
            f[i]=1;
            for(int j=a;j<i;j++)f[i]=add(f[i],mod-mul(f[j],C[i][j]));
        }       
        for(int i=b;i<=m;i++)
        {
            g[i]=1;
            for(int j=b;j<i;j++)g[i]=add(g[i],mod-mul(g[j],C[i][j]));
        }
        int ans=0;
        for(int i=a;i<=n;i++)
            for(int j=b;j<=m;j++)
            {
                int temp=mul(mul(f[i],C[n][i]),mul(g[j],C[m][j]));
                temp=mul(temp,f2[(n-i)*(m-j)]);
                ans=add(ans,temp);
            }
        printf("%d\n",ans);
    }
    return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值