组合数问题

最新推荐文章于 2024-07-25 23:26:35 发布

iamcht

最新推荐文章于 2024-07-25 23:26:35 发布

阅读量307

点赞数 1

文章标签：算法数学数据结构

本文链接：https://blog.csdn.net/weixin_46215084/article/details/112689740

版权

每日一更：组合数问题

- - 题目描述：

题目描述：

组合数 $\tbinom{n}{m}$ 表示的是从 n 个物品中选出 m 个物品的方案数。举个例子，从(1,2,3) 三个物品中选择两个物品可以有 (1,2),(1,3),(2,3)(1,2),(1,3),(2,3) 这三种选择方法。根据组合数的定义，我们可以给出计算组合数 $\tbinom{n}{m}$ 的一般公式：
$\tbinom{n}{m} = \frac{n!}{m!(n - m)!}$
其中 n! = 1 × 2 × ⋯ × n；特别地，0 != 1。
小葱想知道如果给定 n, m 和 k，对于所有的0 ≤ i ≤ n, 0 ≤ j ≤ min(i, m) 有多少对 (i, j) (i, j) 满足k∣ $\tbinom{i}{j}$

输入格式

第一行有两个整数 t, k，其中 t 代表该测试点总共有多少组测试数据，k 的意义见题目描述。
接下来 t 行每行两个整数 n, m，其中 n, m 的意义见题目描述。

输出格式

共 t 行，每行一个整数代表所有的0 ≤ i ≤ n, 0 ≤ j ≤ min(i, m) 有多少对 (i, j) (i, j) 满足k∣ $\tbinom{i}{j}$

输入输出样例

输入 #1

1 2
3 3

输出 #1

输入 #2

2 5
4 5
6 7

输出 #2

0
7

说明/提示

【样例1说明】

在所有可能的情况中，只有 $\binom{2}{1}$ = 2一种情况是 2 的倍数。

【子任务】

对于全部的测试点，保证0 ≤ n, m ≤ 2 × 10³，1 ≤ t ≤ 10⁴.

题目链接：

洛谷：P2822 [NOIP2016 提高组] 组合数问题

分析：

这道题是求组合数里边有多少对数满足其排列组合数（C）能被k整除的总数是多少，假设从n个数中取m个数的排列组合数为s，数学上常用的公式：s = n! / (m! * (n -m)!)但直接用程序暴力输出绝对会tle，这道题有多个测试用例，数据量较大，一定要先打表。
之前我们还学过另外一个有关组合数的公式 $C_{n+1}^{m}$ = $C_{n}^{m}$ + $C_{n}^{m-1}$ 所以，要开一个二维数组，dp打表出每个组合数，a[i][j]代表在i个数中取j个数的排列组合数，状态转移方程：a[i][j] = (a[i - 1][j - 1] + a[i - 1][j]) % k，其实这个状态转移方程就是杨辉三角，先初始化a[0][0] = a[1][1] = a[1][0] = 1，再在每轮循环开始时，a[i][0] = 1，意思是在i个数中取0个数，根据公式 $C_{n}^{m}$ = $C_{n}^{n-m}$ 结果为1，在递推求a数组的时候顺便取模，如果a[i][j] = 0，即说明其被k整除。
如果只处理了这些的话，后续主函数中求a[i][j]有多少个等于0，这道题还是会被卡超时。所以还要用到矩阵前缀和，再开一个s数组，记录前s[i][j]有多少对组合数能被k整除，在循环中加上s[i][j] = s[i - 1][j] + s[i][j - 1] - s[i - 1][j - 1]，再加上本身是否能被k整除。注意在每个j循环结束之后，加上s[i][i + 1] = s[i][i]，否则下一层循环到最后一个数的时候s[i][j] = s[i - 1][j] + s[i][j - 1] - s[i - 1][j - 1]中多减了s[i - 1][j - 1]（因为s[i - 1][j]等于0）。最后，还要注意n < m的情况（被卡了很久），直接输出s[n][n]。

代码如下：

#include <bits/stdc++.h>
using namespace std;
typedef long long LL;
int read(){
	int x, f = 1;
	char ch;
	while(ch = getchar(), ch < '0' || ch > '9') if(ch == '-') f = -1;
	x = ch - '0';
	while(ch = getchar(), ch >= '0' && ch <= '9') x = x * 10 + ch - 48;
	return x * f;
}
LL a[2005][2005], s[2005][2005];
int k;
void u(){
	int i, j;
	a[0][0] = a[1][1] = a[1][0] = 1;
	for(i = 2; i < 2001; i++){
		a[i][0] = 1;
		for(j = 1; j <= i; j++){
			a[i][j] = (a[i - 1][j - 1] + a[i - 1][j]) % k;
			s[i][j] = s[i - 1][j] + s[i][j - 1] - s[i - 1][j - 1];
			if(!a[i][j]) s[i][j]++;
		}
		s[i][i + 1] = s[i][i]; 
	}
} 
int main(){
	int t, n, m;
	t = read();
	k = read();
	u();
	while(t--){
		n = read();
		m = read();
		if(n < m) printf("%lld\n", s[n][n]);
		else printf("%lld\n", s[n][m]);
	} 
	return 0;
}