M - 组合数问题

一、题目：

组合数 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;
}