求组合数
对于组合数的求解,根据公式:C(n, k) = (n * n-1 * ... * n-k) / (k * k-1 * ... * 1)
可以用计算机来模拟这个问题,当n, k很小时,计算机可以正确计算出这个组合数的值。
但是稍微大一点的n, k就会出现爆数据范围的现象有时候int,long long都无法计算出来,这时就会导致计算结果出错。例如:
C(30, 15) = 30 * 29 * ... * 16 / (15 * 14 * ... * 1) 在相乘过程中会出现爆数据范围的情况
我们有两种优化方式:
一:排序交替相乘除
将乘法部分按照升序,除法部分也按照升序,交替相乘
C(30, 15)就变成了:16 * \frac{1}{1} * 17 * \frac{1}{2} * 18 * \frac{1}{3} * ... * 30 * \frac{1}{15}
这里这样可以防止爆数据范围是利用了:连续x个数相乘一定能整除x
二:利用杨辉三角
如下图,行从0行开始,列从0列开始,例如最上面第0行第0列对应a(0, 0)对应1
杨辉三角特点a(n, k) == a(n, n - k), a(n+1, k) == a(n, k-1) + a(n, k)
和组合数是一一对应的。故对于频繁求组合数的题,可以预先处理好这个表,需要使用时直接用
a(i,j)来表示对应的组合数C(i, j)
例如:C(10, 3)对应表中a(10, 3) == 120 // 和数组下标一一对应,从0开始
写一个例题:
分析一下题意,这题就是简单的组合数问题,为什么呢?
机器人从0,0走到n-1, m-1位置总共需要走多少步呢? 一共是 n - 1 + m - 1 = n + m - 2 步
对于样例m = 3, n = 7, 需要走8步,分析一下机器人当前可以存在两种选择:D向下,R向右
对于任意一条路径可以转化为DRDRDRR..的序列,看到这里,就基本知道为什么是组合数问题了
其中D出现2次,R出现6次,相当于求组合数C(8, 2)或者C(8, 6),通过上面的杨辉三角表,我们能迅速得到该值为28
#include<bits/stdc++.h> using namespace std; int main() { int x, y; cin >> x >> y; int n = x + y - 2; // 往右往下一共走了x + y - 2; int a[n+1][n+1]; memset(a, 0, sizeof a); for(int i = 0; i <= n; i++) { a[i][0] = a[i][i] = 1; } for(int i = 0; i <= n; i++) { for(int j = 1; j < i; j++) { a[i][j] = a[i-1][j-1] + a[i-1][j]; } } cout << a[n][y-1] << '\n'; //往右走了y-1次 //cout << a[n][x-1] << '\n'; //往下走了x-1次 return 0; }
#include<bits/stdc++.h> using namespace std; using ll = long long; int main() { int x, y; cin >> x >> y; int n = x + y - 2, k = y - 1; ll ans = 1; for(int i = n, j = 1; j <= k; i--, j++) { ans *= i; ans /= j; } cout << ans; return 0; }
运行结果: