2019牛客暑假多校训练赛第七场 I Chessboard（组合数学＋容斥问题）

题目链接：https://ac.nowcoder.com/acm/contest/887/I

题意：给两个数n,m 在一个无穷大的棋盘上选择一块k*k的正方形区域，在里面的每个方格上都放上至少m个玻璃球，使任意不同行不同列的方格内玻璃球的和都相等且小于等于n，问有几种放玻璃球的方案。

1=<T<=5      1=<n,m<=2000  答案对998244353取模

思路：

上图为题解给的思路，也是这题里很重要的一步转化，此时我们只需要算 f0(k,T)，此时T为减完k*m后的玻璃球总和。

为了更具象地解决问题，我们可以证得如果一个矩阵可以拆分成两个矩阵Ai,Bi，此时Ai同一行的元素相等，Bi同一列的元素相等，那么这个矩阵就是一个有效的矩阵（可以由矩阵加法证得）。此时我们只需要将总和T拆成2k个数，这2k个数分别是Ai中k行的数值以及Bi中k列的数值（Ai,Bi中均最多只有k种不同数值）。于是我们可以初步计算C(2k-1,T+2k-1)【T+2k是因为此时我们需要把T分成2k份，而0的存在比较麻烦，所以我们给2k个数字都加上了1使之可以使用隔板法，其实就是把T分成2k份】。但是算完这个还没结束！因为有重复的情况！我们最开始把矩阵分成了两个矩阵，那么当Ai每列都大于1时，<Ai每行都减一Bi每列都加一> 和原来的Ai,Bi是同一种情况，因为最终矩阵时两个矩阵相加。所以此时我们就减去Ai中每行都大于1的情况即可，所以最终的解是：  此时再对T从0到(n-k*m）遍历计算一下就是最终所有的方案，不要忘记模啊！！

下面是AC代码：

#include<cstdio>
using namespace std;
typedef long long LL;
const int N=2e5+5;
const int mod=998244353;
int n,m,t;
LL f[N],nf[N];
LL quickpow(LL a,LL b)
{
    LL ans=1;
    while(b)
    {
        if(b&1)
            ans=ans*a%mod;
        a=a*a%mod;
        b>>=1;
    }
    return ans;
}
void init()
{
    f[0]=nf[0]=1;
    for(int i=1; i<N; i++)
        f[i]=f[i-1]*i%mod;
    nf[N-1]=quickpow(f[N-1],mod-2);
    for(int i=N-1; i; i--)
        nf[i-1]=nf[i]*i%mod;
}
LL C(int x,int y)
{
    return f[x]*nf[y]%mod*nf[x-y]%mod;
}
int main()
{
    init();
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d%d",&n,&m);
        LL ans=0;
        for(int k=1; k<=n; k++)
        {
            int T=n-k*m;
            for(int i=0; i<=T; i++)
            {
                ans=(ans+C(i+2*k-1,2*k-1))%mod;
                if(i>=k)
                    ans=(ans-C(i+k-1,2*k-1)+mod)%mod;
            }
        }
        printf("%lld\n",ans);
    }
    return 0;
}

 不知不觉坚持下来了第七天！希望以后能继续保持一天一博客，不要做鸽王哈哈 ^ #^