[AcWing] 888. 求组合数 IV(C++实现)求组合数第四种题型(高精度)模板题
求组合数有很多种题型,我们需要根据输入的数据的范围来选哪种方式,具体的方式有如下几种:
主要是询问次数和数据大小的不同,对应了不同的解法
此外,另有高精度组合数和卡特兰数两种特例
星号表示本题的思想。
1. 题目
2. 读题(需要重点注意的东西)
思路:
首先,要知道组合数是什么?公式是什么?
一般地,从a个不同的元素中,任取b(b≤a)个元素为一组,叫作从aa个不同元素中取出b个元素的一个组合,这个组合一共有多少组?
普通公式如下:
本题的主要思路是:
① 分解质因子
② 怎么求a的阶乘中有多少个因子p?(即 ② 怎么求每个质数的次数)
如下图:
(一直加到比 a 的阶乘大为止)
③ 高精度乘法
3. 解法
---------------------------------------------------解法---------------------------------------------------
#include <iostream>
#include <algorithm>
#include <vector>
using namespace std;
const int N = 5010;
int primes[N], cnt;
int sum[N];
bool st[N];
// ① 线性筛,筛质数
void get_primes(int n)
{
for (int i = 2; i <= n; i ++ )
{
if (!st[i]) primes[cnt ++ ] = i;
for (int j = 0; primes[j] <= n / i; j ++ )
{
st[primes[j] * i] = true;
if (i % primes[j] == 0) break;
}
}
}
// ② 计算n的阶乘中素数p的次数
int get(int n, int p)
{
int res = 0;
while (n)
{
res += n / p;
n /= p;
}
return res;
}
// ③ 高精度乘法
vector<int> mul(vector<int> a, int b)
{
vector<int> c;
int t = 0;
for (int i = 0; i < a.size(); i ++ )
{
t += a[i] * b;
c.push_back(t % 10);
t /= 10;
}
while (t)
{
c.push_back(t % 10);
t /= 10;
}
return c;
}
int main()
{
int a, b;
cin >> a >> b;
// ① 分解质因数
get_primes(a);
// ② 计算阶乘中每个素数的次数
for (int i = 0; i < cnt; i ++ )
{
int p = primes[i];
// 用分母的质因子减去分子的质因子就是组合数C的质数的次数
sum[i] = get(a, p) - get(a - b, p) - get(b, p);
}
// ③ 高精度乘法
vector<int> res;
res.push_back(1);
for (int i = 0; i < cnt; i ++ )
for (int j = 0; j < sum[i]; j ++ )
res = mul(res, primes[i]);
// 输出该值
for (int i = res.size() - 1; i >= 0; i -- ) printf("%d", res[i]);
puts("");
return 0;
}
4. 可能有帮助的前置习题
5. 所用到的数据结构与算法思想
- 组合数
- 线性筛
- 高精度乘法
6. 总结
求组合数第四种题型(高精度)的模板题,理解思路并背下代码。