Lucas定理 原理、证明及代码模板

一、定理

对于素数p，及正整数a、b，当求组合数$C_a^b$对p取余时，如果a和b很大$(1\le b\le a\le 10^{15})$，而p相对较小$(1\le p\le 10^5)$时，我们用Lucas定理来求。

结论：$C_a^b\equiv C_{\frac{a}{p}}^{\frac{b}{p}}\cdot C_{a\%p}^{b\%p}$ (mod p)

---

二、证明

首先将a和b用p进制表示出来
$$a=a_0{p}^0+a_1{p}^1+...+a_k{p}^k$$$$b=b_0{p}^0+b_1{p}^1+...+b_k{p}^k$$

然后，在证明前，我们要用到两个引理。
（1）p是素数，$p|C_p^k$，也就是说p是$C_p^k$的一个因子。这个很好证明，因为$C_p^k=\frac{p!}{k!(p-k)!}$分子p的阶乘中含有p作为因子，因为p是素数，所以分母中的阶乘都和p互素，那么p就是$C_p^k$的一个因子了，即p能整除$C_p^k$。
（2）$(x+y{)}^p\equiv x^p+y^p$ (mod p)。这个也很好证明，根据二项式定理将括号拆开可以得$(x+y{)}^p=x^p+{\sum }_{k=1}^{p-1}{C}_p^k{x}^k{y}^{p-k}+y^p$，而中间项含有$C_p^k$，是p的整数倍，因此在模p的情况下都可以约掉。就剩下了两边的两项。同理可得：$(x+y{)}^{p^k}\equiv x^{p^k}+y^{p^k}$ (mod p)

有了这些准备之后，我们再来证明，根据二项式定理将$(1+x{)}^a$展开。因为a可以用p进制表示，所以
$$(1+x{)}^a=(1+x{)}^{a_0{p}^0+a_1{p}^1+...+a_k{p}^k}$$$$(1+x{)}^a=(1+x{)}^{a_0}\times (1+x{)}^{a_{1}p}\times ...\times (1+x{)}^{a_k{p}^k}$$$$(1+x{)}^a\equiv (1+x{)}^{a_0}\times (1+x^p{)}^{a_1}\times ...\times (1+x^{p^k}{)}^{a_k}$$ $$(1+x{)}^a\equiv \sum _{m=0}^{a_0}{C}_{a_0}^m{x}^m\times \sum _{m=0}^{a_1}{C}_{a_1}^m{x}^{pm}\times ...\times \sum _{m=0}^{a_k}{C}_{a_k}^m{x}^{p^{k}m}$$然后我们再看这个同余式左右两边的$x^b$的系数。左边$x^b$的系数是$C_a^b$。再看右边，右边所有x的幂都和p进制下的展开是一模一样的，而正好b的p进制是$b=b_0{p}^0+b_1{p}^1+...+b_k{p}^k$，并且右边的每一项都是乘起来的，因此要找$x^n$这一项，当且仅当第一个的i取$b_0$，第二个i取$b_1$，…，第k项的i取$b_k$（他们加起来刚好是b的p进制展开，所以幂次就是b），而他们的每一项的系数就是$C_{a_0}^{b_0}$、$C_{a_1}^{b_1}$、…$C_{a_k}^{b_k}$。因此$x^n$这一项的系数就是将他们乘起来即$C_{a_0}^{b_0}\times C_{a_1}^{b_1}\times ...\times C_{a_k}^{b_k}$ (mod p)。

因此得证，根据左右两边$x^n$的系数相同可得：$C_a^b\equiv C_{a_0}^{b_0}\times C_{a_1}^{b_1}\times ...\times C_{a_k}^{b_k}$ (mod p)。时间复杂度大概是$O(p\log N\log p)$。

---

三、代码模板

#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
typedef long long LL;

int p,n;

int qmi(int a,int b)	//快速幂求逆元
{
    LL res = 1;
    while(b)
    {
        if(b & 1)
            res = (LL)res * a % p;
        a = (LL)a * a % p;
        b >>= 1;
    }
    return res;
}

int C(int a,int b)	//公式求C(a,b)
{
    int res = 1;
    for(int i = 1,j = a;i <= b;i++,j--)
    {
        res = (LL)res * j % p;		//分子从a-b+1乘到a
        res = (LL)res * qmi(i,p-2) % p;	//分母是b的阶乘，转化为乘以阶乘的逆元
    }
    return res;
}

int lucas(LL a,LL b)	//Lucas定理
{
    if(a <p && b < p)	//如果a和b都小于p，可以直接用公式法求c
        return C(a,b);
    return (LL)C(a%p,b%p)*lucas(a/p,b/p) % p;	//否则就用Lucas定理，用Lucas结论来求解
}

int main()
{
    cin >> n;
    while (n -- )
    {
        LL a,b;
        cin >> a >> b >> p;		//输入a、b和模数p
        cout << lucas(a,b) << endl;
    }
    
    return 0;
}