一、题目:
组合数 mn) 表示的是从 n 个物品中选出 m 个物品的方案数。举个例子,从 (1,2,3)(1,2,3) 三个物品中选择两个物品可以有 (1,2),(1,3),(2,3)(1,2),(1,3),(2,3) 这三种选择方法。根据组合数的定义,我们可以给出计算组合数 (mn) 的一般公式:
(mn)=m!(n−m)!n!
其中!=1×2×⋯×n!=1×2×⋯×n;特别地,定义 0!=10!=1。
小葱想知道如果给定 n,m 和 k。
输入格式
第一行有两个整数 t,k,其中 t 代表该测试点总共有多少组测试数据,k 的意义见问题描述。
接下来 t 行每行两个整数 n,m,其中n,m 的意义见问题描述。
输出格式
共 t 行,每行一个整数代表所有的 0≤i≤n,0≤j≤min(i,m) 中有多少对 (i,j) 满足 k∣(ji)。
输入输出样例
输入 #1复制
1 2 3 3
输出 #1复制
1
输入 #2复制
2 5 4 5 6 7
输出 #2复制
0 7
说明/提示
【样例1说明】
在所有可能的情况中,只有 (21)=2(12)=2 一种情况是 22 的倍数。
【子任务】
- 对于全部的测试点,保证 0≤�,�≤2×1030≤n,m≤2×103,1≤�≤1041≤t≤104。
二、思路:
c[i][j]即为从i件物品中选j件的方案数。如果第i件物品不选,方案数就变为c[i-1][j],如果选第i件物品,方案数就变为c[i-1][j-1],总方案数就为两种情况的方案数之和。
三、代码:
#include<cstdio>
#include<iostream>
#define int long long
using namespace std;
int ans[10000][10000];
int c[10000][10000];
int t,k,n,m;
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;//别忘取模
ans[i][j] = ans[i - 1][j] + ans[i][j - 1] - ans[i - 1][j - 1];
if (!c[i][j])ans[i][j]++;
}
ans[i][i + 1] = ans[i][i];
}
}
void solve()
{
cin >> t >> k;
build();
while (t--)
{
cin >> n >> m;
if (m > n)cout << ans[n][n] << endl;
else cout << ans[n][m] << endl;
}
}
signed main() {
solve();
return 0;
}