AcWing 1316. 有趣的数列(卡特兰数)

一.题目链接

二.思路

卡特兰数的应用场景

三.代码

#include <iostream>
#include <cstring>
#include <algorithm>

using namespace std;
typedef long long ll;
const int N = 2000010;

int primes[N], cnt;
bool st[N];

int n, p;

int qmi(int a, int b, int c)
{
    int res = 1;
    while(b)
    {
        if(b & 1) res = (ll)res * a % c;
        a = (ll)a * a % c;
        b >>= 1;
    }
    return res;
}
void init(int n)
{
    for(int i = 2; i <= n; i ++)
    {
        if(!st[i]) primes[cnt ++] = i;
        for(int j = 0; i * primes[j] <= n; j ++)
        {
            st[i * primes[j]] = true;
            if(i % primes[j] == 0) break;
        }
    }
}
//n的阶乘当中含有多少个质因数p
int get(int n, int p)
{
    int s = 0;
    while (n){
        s += n / p;
        n /= p;
    }
    return s;
}
int C(int a, int b)
{
    int res = 1;
    for(int i = 0; i < cnt; i ++)
    {
        int pm = primes[i];//质数p
        int s = get(a, pm) - get(b, pm) - get(a - b, pm);
         while(s --) res = (ll)res * pm % p;//边算边加
       //res = (ll)res * qmi(pm, s, p) % p;
    }
    return res;
}
int main()
{
    cin >> n >> p;
    init(n * 2);//先预处理质数
    //答案为卡特兰数C(2n,n) - C(2n, n - 1);
    cout << (C(2 * n, n) - C(2 * n, n - 1) + p) % p << endl;
    return 0;   
}

四.总结

• 当求组合数，数据规模较大时（比如大于等于5000时），需要用质因数分解的方法求解组合数。
  1.若需要mod一个数。则直接预处理质数，阶乘分解质因数即可。
  2.若不mod一个数。则说明肯定会爆longlong,还需将相关操作改成高精度算法。