题目描述
组合数 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;
}