洛谷P2822 组合数问题

题目描述
组合数 Cmn（m在上，n在下）表示的是从 n个物品中选出 m 个物品的方案数。举个例子，从 (1,2,3) 三个物品中选择两个物品可以有 (1,2),(1,3),(2,3) 这三种选择方法。根据组合数的定义，我们可以给出计算组合数 Cmn的一般公式：

​
其中 n=1×2×⋯×n；特别地，定义 0!=1。
小葱想知道如果给定 n,m和 k，对于所有的 0\leq i\leq n,0\leq j\leq \min \left ( i, m \right )0≤i≤n,0≤j≤min(i,m) 有多少对 (i,j)(i,j) 满足 C_i^jC
i是 k 的倍数。
输入输出格式
输入格式：
第一行有两个整数 t,k，其中 t 代表该测试点总共有多少组测试数据，k 的意义见问题描述。
接下来 t 行每行两个整数 n,m，其中 n,m 的意义见问题描述。
输出格式：
共 t行，每行一个整数代表所有的0 ≤ i ≤ n,0 ≤ j ≤ min(i,m) 中有多少对 ( i, j ) 满足 C_i^j是 k 的倍数。
输入输出样例
输入样例#1：
1 2
3 3
输出样例#1：
1
输入样例#2：
2 5
4 5
6 7
输出样例#2：
0
7
说明

【样例1说明】

在所有可能的情况中，只有C_2^1 =2是2的倍数。
2
1
​
【子任务】

这道题需要用到杨辉三角形（但是谁又能在考场上想到它呢…)。同时为了防止杨辉三角中的数过大，你还要记得随时给它取模。
但是事情还没有结束，为了能降低时间复杂度，我们需要把之前的o(n)修改变成o(1)，那就用二维前缀和维护吧。

#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
using namespace std;
int T,k;
int s[2003][2003],c[2003][2003];
//C[m][n] = c[n - m][n]	  
int main()
{
	scanf("%d%d",&T,&k);
	c[1][1] = 1;
	for(int i = 0;i <= 2000;i++)c[i][0] = 1;
	for(int i = 0;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 = 2;i <= 2000;i++)
	{
		for(int j = 1;j <= i;j++)
		{
			s[i][j] = (s[i - 1][j]+s[i][j - 1]-s[i - 1][j - 1]);//求前缀和 
			if(c[i][j] == 0)s[i][j] += 1;//如果这个方案数可以被k整除，那结果就加1啦 
		}
		s[i][i+1] = s[i][i];//更新梁辉三角的右边 
	}	
	while(T--)
	{
		int n,m;
		scanf("%d%d",&n,&m);
		if(m > n)m = n; 
		printf("%d\n",s[n][m]);
	}
	return 0;
}