没想到这道题竟然这么水…
我们发现m,n都非常小,完全可以 O ( n m ) O(nm) O(nm)预处理出stripe数组,即代表 ( i , j ) (i,j) (i,j)
及其向上的一列的个数,然后进行递推即可。
#include<cstdio>
#include<algorithm>
using namespace std;
const int maxn = 2003;
int C[maxn][maxn];
int ans[maxn][maxn], stripe[maxn][maxn];
int main()
{
//froepen("in.txt","r",stdin);
int T,k;
scanf("%d%d",&T,&k);
for(int i = 1;i <= 2001; ++i) C[i][i] = C[i][0] = 1;
for(int i = 1;i <= 2001; ++i)
for(int j = 1;j < i; ++j)
C[i][j] = (C[i-1][j] + C[i-1][j-1]) % k;
for(int i = 2;i <= 2001; ++i)
for(int j = 0;j <= i; ++j)
{
if(C[i][j] == 0) ans[i][j] = stripe[i][j] = 1;
stripe[i][j] += stripe[i-1][j];
ans[i][j] += stripe[i-1][j];
if(j > 0) ans[i][j] += ans[i][j-1];
}
while(T--)
{
int n,m;
scanf("%d%d",&n,&m);
m = min(m,n);
printf("%d\n",ans[n][m]);
}
return 0;
}