C++求解组合数

递推法 杨辉三角

递推式$C_n^m=C_{n-1}^m+C_{n-1}^{m-1}$
从 n个不同的数中选出 m个的方案数是 $C_n^m$
对第 1 个数有 选或不选 两种决策： $C_n^m$
若不选，则从剩下的 n−1个中选 m 个，即 ​； $C_{n-1}^m$
若选，则从剩下的 n−1个中选 m−1 个，即​。$C_{n-1}^{m-1}$
加法原理$C_n^m=C_{n-1}^m+C_{n-1}^{m-1}$

void comb()
{
    for(int i=0;i<n;i++)
    {
        for(int j=0;j<=i;j++)
        {
            if(j==0)C[i][j] = 1;
            else C[i][j]=(C[i-1][j]+C[i-1][j-1])%mod;
        }
    }
}

快速幂+逆元

组合数计算优化方案

递推法 $C_n^m=C_{n-1}^m+C_{n-1}^{m-1}$，会 TLE。
考虑用 $C_n^m=\frac{n!}{(n-m)!\cdot m!}$ 直接计算。
开两个数组分别存模意义下的阶乘和阶乘的逆元：

• 用 $f[x]$ 存 $x!\mathrm{mod}p$ 的值；
• 用 $g[x]$ 存 $(x!{)}^{-1}\mathrm{mod}p$ 的值。
  因为 $p$ 是质数且 $n,m$ 都小于 $p$，即 $n,m$ 与 $p$ 互质，
  所以根据费马小定理 $a\cdot a^{p-2}\equiv 1(\mathrm{mod}p)$，
  因为$a\cdot a^{-1}\equiv 1(\mathrm{mod}p)$
  所以 $a^{-1}\equiv a^{p-2}(\mathrm{mod}p)$
  可以用快速幂求逆元。
  无法直接计算 $\frac{n!}{(n-m)!\cdot m!}\mathrm{mod}p$，所以转化为 $n!\cdot ((n-m)!\cdot m!{)}^{-1}\mathrm{mod}p$。
  所以$C_n^m\mathrm{mod}p=f[n]\times g[n-m]\times g[m]\mathrm{mod}p$。
  

typedef long long ll;
ll qpow(ll a,int b,ll p)
{
   ll res;
   while(b)
   {
    if(a & 1)res=res*a%p;
    a=a*a%p;
    b >>= 1;
   }
   return res;
}
void init()
{
    f[0]=1,g[0]=1;
    for(int i=1;i<n;i++)
    {
        f[i] = i*f[i-1]%p;
        g[i] = g[i-1]*qpow(i,p-2)%p;
    }
}
ll comb(ll n,ll m)
{
    return f[n]*g[n-m]%p*g[m]%p;
}

卢卡斯定理

卢卡斯定理：
$C_n^m\equiv C_{n/p}^{m/p}\cdot C_{n\mathrm{mod}p}^{m\mathrm{mod}p}(\mathrm{mod}p)$
其中 $p$ 为质数。
$n\mathrm{mod}p$ 和 $m\mathrm{mod}p$ 一定是小于 $p$ 的数，可以直接求解，
$C_{n/p}^{m/p}$ 可以继续用 Lucas 定理求解。
边界条件：当 $m=0$ 时，返回 $1$。

typedef long long ll;
ll qpow(ll a,int b,ll p)
{
   ll res;
   while(b)
   {
    if(a & 1)res=res*a%p;
    a=a*a%p;
    b >>= 1;
   }
   return res;
}
void init(ll N,ll p)
{
    f[0]=1,g[0]=1;
    for(int i=1;i<N;i++)
    {
        f[i] = i*f[i-1]%p;
        g[i] = g[i-1]*qpow(i,p-2)%p;
    }
}
ll comb(ll n,ll m,ll p)
{
    return f[n]*g[n-m]%p*g[m]%p;
}
ll lucas(ll n,ll m)
{
    if(m==0)return 1;
    return lucas(n/p,m/p)*comb(n%p,m%p)%p;
}