AcWing 888. 求组合数 IV（高精度算法）

AcWing 888. 求组合数 IV

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

输入格式
共一行，包含两个整数a和b。

输出格式
共一行，输出Cba的值。

数据范围
1≤b≤a≤5000
输入样例：

5 3

输出样例：

10

这道题用到了高精度运算，可以参考一下我的以前的博文，我们不是简简单单的相乘就可以了，这样时间复杂度会达到o(n²根2）级别这样不是很好，y总是先求结果的质因数，然后逐步乘以质因数，这个思路还是非常非常很好的，我们求a!的质因数的方法也不是简简单单的试除法，有一个式子，我们可以相应的求出a!中质数p的个数。式子如下

这样就很轻松的求出a!中p的个数了。
然后我们去进行一个相减操作。最后求出相应质因数的个数
代码如下

#include<iostream>
#include<algorithm>
#include<vector>

using namespace std;


const int N=5050;
int cnt;
int primes[N],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;
        }
    }
}

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()||t;i++)
    {
        if(i<a.size()) t+=a[i]*b;
        c.push_back(t%10);
        t/=10;
    }
    while(c.back()==0&&c.size()>1) c.pop_back();
    return c;
}


int main(void)
{
    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> s;
    s.push_back(1);
    for(int i=0;i<cnt;i++)
    {
        for(int j=0;j<sum[i];j++)
        s=mul(s,primes[i]);
    }
    for(int i=s.size()-1;i>=0;i--)
    printf("%d",s[i]);
}