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

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

 

题意：给定n，m要求在一个k*k的正方形内放一些球，正方形里每个1*1的格子至少放m个球，且任意不同行不同列的球的和是相等的且小于等于n，问可行的任意k*k的正方形有多少。

数据范围：1<=n,m<=2000

 

思路： 我们计“选择大小为k∗k的棋盘，每个方格放的玻璃球数都不小于m且每不同行不同列的方格内的玻璃球数量的总和均为T ”的方案数为，那么很显然的总的方案数即为:,那么接下来只要考虑如何求解,首先我们可以把此题的一个正方形矩阵拆成两个正方形的矩阵，然后往第一个矩阵中一行一行的加数（一行都是同一个数），第二个矩阵中一列一列的加数（一列都是同一个数），然后把两个矩阵对应相加得到的矩阵就是一个合法答案。所以现在的问题就变成了对第一个矩阵的k行加数（行可以是0），对第二个矩阵的k列加数（列可以是0），那么现在就是把T个数分给2*k个位置，每个位置可以为零，那么这就是高中的隔板法，因为每一个位置可以是0，所以就要加入2*k个新球（这个高中有讲），对新生成的T+2*k分出2*k块，所以只需要在T+2*k-1个缝隙中插2*k-1个板就可以完成。接下来考虑去重，当第一个矩阵的每一行都插入了至少1，那么把第一个矩阵每一行都减1对第二个矩阵每一列都多插入1最后两个矩阵加和得到的情况是一样的，而这种情况都出现在第一个矩阵的k行每一行至少是1的情况下，那么减掉这些就可以了，而此时隔板的时候只有第二个矩阵可以每列是0，所以对于T个球新加k个球就可以了，此时就是在T+k-1个缝隙中插入2*k-1个板就可以了。所以答案就是

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=2e5+5;
const int mod=998244353;
int n,m,t;
ll fac[N],facv[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(){
    fac[0]=facv[0]=1;
    for(int i=1;i<N;i++)fac[i]=fac[i-1]*i%mod;
    facv[N-1]=quickpow(fac[N-1],mod-2);
    for(int i=N-1;i;i--)facv[i-1]=facv[i]*i%mod;
}
ll C(int x,int y){
    return fac[x]*facv[y]%mod*facv[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;
}