组合数计算方法

目录

公式计算
对数计算
递推计算
逆元计算
卢卡斯定理

前言

公式计算

直接用组合数的公式
$$C_m^n=\frac{m!}{n!(m-n)!}$$
但是很容易溢出，$21!=51090942171709440000$

对数计算

利用取对数，求前缀和
$$\ln C_m^n=\ln m!-\ln n!-\ln (m-n)!=\sum _{x=1}^m\ln x-\sum _{x=1}^n\ln x-\sum _{x=1}^{m-n}\ln x$$
用$O(n)$的时间来求$\ln x$的前缀，但是可能有浮点误差。

递推计算

利用递推式$$C_m^n=C_{m-1}^{n-1}+C_{m-1}^n$$
复杂度$O(n^2)$，不算太好

逆元计算

通常题目会告诉一个模数 $p$，这个时候就可以用乘法逆元把除法转化成来计算，于是回到最开始的式子$$C_m^n=\frac{m!}{n!(m-n)!}$$
把除法转化成乘法逆元$$C_m^n\equiv m!\cdot \mathrm{i}\mathrm{n}\mathrm{v}(n!)\cdot \mathrm{i}\mathrm{n}\mathrm{v}[(m-n)!](\mathrm{m}\mathrm{o}\mathrm{d}p)$$
这样就可以$O(lo{g}_n)$求逆元和$O(n)$预处理，但是遇到$m>p$的情况就不能保证$n$和$m-n$的逆元存在了（它们可能是$p$的倍数）
$ps$: 对于$A\ast x=1(\mathrm{m}\mathrm{o}\mathrm{d}p)$,称$A$关于$1$模$p$的乘法逆元为$x$，又通过费马小定理$A^{p-1}\equiv 1(\mathrm{m}\mathrm{o}\mathrm{d}p)$，可得$x=A^{p-2}(\mathrm{m}\mathrm{o}\mathrm{d}p)$

卢卡斯定理

对于非负整数$m,n$和质数$p$，$C_m^n\equiv {\prod }_{i=0}^k{C}_{m_i}^{n_i}(\mathrm{m}\mathrm{o}\mathrm{d}p)$，其中
$m=m_k{p}^k+\cdots +m_{1}p+m_0$
$n=n_k{p}^k+\cdots +n_{1}p+n_0$
是$n,m$的$p$进制展开

通常使用如下公式计算$C_m^n$
$$C_m^n\equiv C_{m\mathrm{m}\mathrm{o}\mathrm{d}p}^{n\mathrm{m}\mathrm{o}\mathrm{d}p}\cdot C_{\lfloor m/p\rfloor }^{\lfloor n/p\rfloor }(\mathrm{m}\mathrm{o}\mathrm{d}p)$$
注意：当$m<n$时，规定$C_m^n=0$
使用上面的公式就可以利用递归求解，当n=0时退出。
上板子

// 需要先预处理出阶乘fact[]，利用快速幂求逆元inv
inline ll C(ll m, ll n, ll p)
{
    return m < n ? 0 : fact[m] * inv(fact[n], p) % p * inv(fact[m - n], p) % p;
}
inline ll lucas(ll m, ll n, ll p)
{
    return n == 0 ? 1 % p : lucas(m / p, n / p, p) * C(m % p, n % p, p) % p;
}

逆元+预处理，复杂度是$O(p+lo{g}_{p}m)$，$10^6$没问题。
证明过程