预处理组合数和逆元o(n)

顾客言

于 2024-02-15 19:39:17 发布

阅读量160

点赞数 2

文章标签： java 算法数据结构

本文链接：https://blog.csdn.net/m0_73673533/article/details/136122712

版权

int fact[N], infact[N];
int qpow(int a, int b)
{
    int res = 1;
    while (b)
    {
        if (b & 1)
            res = res * a % mod;
        a = a * a % mod;
        b >>= 1;
    }
    return res;
}
void init()
{
    fact[0] = 1;
    for (int i = 1; i <= N; i++)
        fact[i] = fact[i - 1] * i % mod;
    infact[N] = qpow(fact[N], mod - 2);
    for (int i = N; i >= 1; i--)
        infact[i - 1] = infact[i] * i % mod;
}
int C(int n, int m)
{
    return fact[n] * infact[m] % mod * infact[n - m] % mod;
}