【问题描述】
组合数表示的是从n个物品中选出m个物品的方案数。举个例子,从(1,2,3) 三个物品中选择两个物品可以有(1,2),(1,3),(2,3)这三种选择方法。根据组合数的定义,我们可以给出计算组合数的一般公式:
其中n! = 1 × 2 × … × n。
小葱想知道如果给定n,m和k,对于所有的0 <= i <= n,0 <= j <= min(i,m)有多少对 (i,j)满足是k的倍数。
【输入格式】
第一行有两个整数t,k,其中t代表该测试点总共有多少组测试数据,k的意义见 【问题描述】。
接下来t行每行两个整数n,m,其中n,m的意义见【问题描述】。
【输出格式】
t行,每行一个整数代表所有的0 <= i <= n,0 <= j <= min(i,m)有多少对 (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的倍数
题解:这个题目是求前缀和,我们用杨辉三角求2000内的组合数,因为组合数太大,所有我们对k取模,对k取模不会影响它是不是k的倍数,然后预处理前缀和就好,上代码。
#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
int a[2005][2005];
int dp[2005][2005];
void init(int k)
{
for(int i=0;i<=2000;i++)
{
a[i][0]=1;
a[i][i]=1;
}
for(int i=2;i<=2000;i++)
{
for(int j=1;j<i;j++)
{
a[i][j]=(a[i-1][j-1]+a[i-1][j]);
if(a[i][j]!=0)
{
a[i][j]=a[i][j]%k;
if(a[i][j]==0)
a[i][j]=k;
}
}
}
//预处理前缀和,dp[i][j]代表i行j列
dp[0][0]=0;
for(int i=1;i<=2000;i++)
{
dp[i][0]=dp[i-1][0]+(a[i][0]%k==0&&a[i][0]!=0);
dp[0][i]=dp[0][i-1]+(a[0][i]%k==0&&a[0][i]!=0);
}
for(int i=1;i<=2000;i++)
{
for(int j=1;j<=2000;j++)
{
dp[i][j]=dp[i-1][j]+dp[i][j-1]-dp[i-1][j-1]+(a[i][j]%k==0&&a[i][j]!=0);
}
}
}
int main()
{
int t,k,n,m;
cin>>t>>k;
init(k);
while(t--)
{
scanf("%d%d",&n,&m);
printf("%d\n",dp[n][m]);
}
}