题目链接: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;
}