AcWing 888 求组合数 IV

题目描述：

输入a,b，求C(a,b)的值。注意结果可能很大，需要使用高精度计算。

输入格式

共一行，包含两个整数a和b。

输出格式

共一行，输出C(a,b)的值。

数据范围

1≤b≤a≤5000

输入样例：

5 3

输出样例：

10

分析：

本题要求组合数的实际值，不能对结果取模，所以要使用高精度。C(a,b) = a! / (b! * (a - b)!)。为了避免除法运算，我们对分子分母都进行质因子分解，需要解决的问题是如何快速的求出一个数的阶乘中包含的各个质因子的次数。首先需要用线性筛法筛出我们需要的素数，然后在1到n中，要想求出包含有多少个质因子p，我们知道，所有p的倍数至少包含一个p因子，所有p^2的倍数至少包含两个p因子，...，故n！中质因子p的个数为n / p + n / p^2 +...。C(a,b)中包含p的个数是a!中p的个数减去b!中p的个数再减去（a-b）!中p的个数。最后将分解出的各个质因子使用高精度乘法乘到一起就是答案了。

#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
const int maxn = 5005;
int primes[maxn],sum[maxn],cnt;
bool st[maxn];
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[i * primes[j]] = true;
            if(i % primes[j] == 0)  break;
        }
    }
}
int get(int n,int p){//前n个数中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 = 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];
        sum[i] = get(a,p) - get(b,p) - get(a - b,p);
    }
    vector<int> res(1,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;
}