线性递推组合数,然后求一个前缀和。
对着大佬题解改改改一直没对,然后发现m可以大于n……
#include<cstdio>
const int N=2e3+10;
int t,k,n,m;
long long C[N][N],sum[N][N];
int main()
{
//freopen("in.txt","r",stdin);
scanf("%d%d",&t,&k);
for(int i=0;i<=2000;i++)C[i][0]=1;
C[1][1]=1;
for(int i=2;i<=2000;i++)
for(int j=1;j<=i;j++)
C[i][j]=(C[i-1][j]+C[i-1][j-1])%k;
for(int i=0;i<=2000;i++)
{
for(int j=1;j<=i;j++)
{
sum[i][j]=sum[i-1][j]+sum[i][j-1]-sum[i-1][j-1];
if(C[i][j]==0)sum[i][j]++;
}
sum[i][i+1]=sum[i][i];
}
while(t--)
{
scanf("%d%d",&n,&m);if(m>n)m=n;
printf("%lld\n",sum[n][m]);
}
return 0;
}
总结
考察组合数的递推