题目:
http://acm.zzu.edu.cn/problem.php?cid=1048&pid=7
问题 H: 「NOIP2016」组合数问题
时间限制: 1 Sec 内存限制: 512 MB
提交: 23 解决: 9
[提交] [状态] [讨论版] [命题人:外部导入]
题目描述
组合数表示的是从 n nn 个物品中选出 m mm 个物品的方案数。举个例子,从 (1,2,3) (1, 2, 3)(1,2,3) 三个物品中选择两个物品可以有 (1,2) (1, 2)(1,2),(1,3) (1, 3)(1,3),(2,3) (2, 3)(2,3) 这三种选择方法。
根据组合数的定义,我们可以给出计算组合数的一般公式:
Cnm=n!m!(n−m)! C_n ^ m = \frac{n!}{m!(n - m)!}C
n
m
m!(n−m)!
n!
其中 n!=1×2×⋯×n n! = 1 \times 2 \times \cdots \times nn!=1×2×⋯×n。
小葱想知道如果给定 n nn,m mm 和 k kk,对于所有的 0≤i≤n 0 \leq i \leq n0≤i≤n,0≤j≤min(i,m) 0 \leq j \leq \min(i, m)0≤j≤min(i,m) 有多少对 (i,j) (i, j)(i,j) 满足是 k kk 的倍数。
输入格式
第一行有两个整数 t tt,k kk,其中 t tt 代表该测试点总共有多少组测试数据,k kk 的意义见 「问题描述」。
接下来 t tt 行每行两个整数 n nn,m mm,其中 n nn,m mm 的意义见「问题描述」。
输出格式
t tt 行,每行一个整数代表答案。
样例
样例输入 1
1 2
3 3
样例输出 1
1
样例输入 2
2 5
4 5
6 7
样例输出 2
0
7
数据范围与提示
3≤n,m≤2000,2≤k≤21,1≤1≤10000 3 \leq n, m \leq 2000, 2 \leq k \leq 21, 1 \leq 1 \leq 100003≤n,m≤2000,2≤k≤21,1≤1≤10000
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
int c[2001][2001],ans[2001][2001];
int t,k;
void build()
{
c[0][0]=1;
c[1][0]=c[1][1]=1;
for(int i=2;i<=2000;i++)
{
c[i][0]=1;
for(int j=1;j<=i;j++)
{
c[i][j]=(c[i-1][j-1]+c[i-1][j])%k;//杨辉三角+直接%k判断是否为k的整数倍
ans[i][j]=ans[i-1][j]+ans[i][j-1]-ans[i-1][j-1];//前缀和 -a[i][j]
if(!c[i][j]) ans[i][j]++;//加上a[i][j] 得到前缀和ans[i][j]
}
ans[i][i+1] =ans[i][i];//在第16行那一步时,需要用到对角线旁边这个元素,所以需要赋值
}
}
int main()
{
cin>>t>>k;
build();
while(t--)
{
int n,m;
cin>>n>>m;
if(m>n) m=n;
cout<<ans[n][m]<<endl;
}
return 0;
}