Lucas定理和扩展

Lucas定理是用来求 c(n,m) mod p，p为素数的值。
它是用来求组合数 C(n, m) % p 值的定理，这里的p是素数。所以，它是一个解决大组合数求模的算法。

卢卡斯定理适用于模数 𝑝 为质数的情况

首先我们知道组合数的计算公式

𝐶𝑚𝑛=𝑛!/𝑚!(𝑛−𝑚)!

所以我们可以先递推出阶乘

再用快速幂或 𝐸𝑋𝐺𝐶𝐷 计算逆元

也可以直接递推阶乘逆元然后就可以计算 𝐶了

ll qpow(ll a,int b,int p)
{
    ll ans=1,base=a%p;
    while(b)
    {
        if(b&1) ans=ans*base%p;
        base=base*base%p;
        b>>=1;
    }
    return ans;
}

ll Lucas(ll n,ll m,int p)
{
    if(!m) return 1;
    return C(n%p,m%p,p)*Lucas(n/p,m/p,p)%p;
}

ll jc[maxn];

ll C(ll n,ll m,int p)
{
    if(n<m) return 0;
    return (jc[n]*qpow(jc[m],p-2,p)%p)*qpow(jc[n-m],p-2,p)%p;
}

另一个版本

// mod = 100007
int Lucas (ll n , ll m , int p) 
{
  return m == 0 ? 1 : 1ll*comb (n%p , m%p , p) * Lucas (n/p,m/p,p)%p ;
}
//comb()函数中，因为q ， r < p , 所以这部分暴力完成即可。
//C++求C(n, m) mod 10007    版本二 要求p z在100000左右
ll f[N];
void init(int p) 
{       //f[n] = n!
    f[0] = 1;
    for (int i=1; i<=p; ++i) f[i] = f[i-1] * i % p;
} 
ll pow_mod(ll a, ll x, int p)
{
    ll ret = 1;
    while (x)
        {
        if (x & 1)  ret = ret * a % p;
        a = a * a % p;
        x >>= 1;
    }
    return ret;
}
  
ll Lucas(ll n, ll k, int p)        //C (n, k) % p
{
     ll ret = 1;
     while (n && k) 
        {
        ll nn = n % p, kk = k % p;
        if (nn < kk) return 0;  //inv (f[kk]) = f[kk] ^ (p - 2) % p
        ret = ret * f[nn] * pow_mod (f[kk] * f[nn-kk] % p, p - 2, p) % p;
        n /= p, k /= p;
     }
     return ret;
}
int main(void)
{
    init (p);
    printf ("%I64d\n", Lucas (n, m, p));
    return 0;
}