OI模板 卢卡斯定理

卢卡斯定理 / $\text{Lucas}$定理

现有问题：

给定整数 $n,m,p$，求 $\left(\genfrac{}{}{0ex}{}{n}{m}\right)\mathrm{m}\mathrm{o}\mathrm{d}p$，保证 $p\in \mathbb{P}$。

$1\le n,m,p\le 10^5$。

可以用 Lucas 定理解决。

定义

对于 $p\in \mathbb{P}$，有 $\left(\genfrac{}{}{0ex}{}{n}{m}\right)\mathrm{m}\mathrm{o}\mathrm{d}p=\left(\genfrac{}{}{0ex}{}{\lfloor n/p\rfloor }{\lfloor m/p\rfloor }\right)\times \left(\genfrac{}{}{0ex}{}{n\mathrm{m}\mathrm{o}\mathrm{d}p}{m\mathrm{m}\mathrm{o}\mathrm{d}p}\right)\mathrm{m}\mathrm{o}\mathrm{d}p$

另一种形式：$\left(\genfrac{}{}{0ex}{}{n}{m}\right)\mathrm{m}\mathrm{o}\mathrm{d}p=\prod _{i=1}^k\left(\genfrac{}{}{0ex}{}{a_i}{b_i}\right)\mathrm{m}\mathrm{o}\mathrm{d}p$（$a,b$ 分别为 $n,m$ 在 $p$ 进制下的某一位，$k$ 是它们 $p$ 进制位数的较大值）。

证明

以下的 $p\in \mathbb{P}$。

$$\because \left(\genfrac{}{}{0ex}{}{p}{n}\right)\mathrm{m}\mathrm{o}\mathrm{d}p=\frac{p!}{n!(p-n)!}=[n=0\vee n=p]$$

$$\begin{array}{rl}\therefore (a+b{)}^p& =\sum _{n=0}^p\left(\genfrac{}{}{0ex}{}{p}{n}\right)a^n{b}^{p-n}\\ & \equiv \sum _{n=0}^p[n=0\vee n=p]a^n{b}^{p-n}\\ & \equiv a^p+b^p(\mathrm{m}\mathrm{o}\mathrm{d}p)\end{array}$$

上面的结论可以用费马小定理推，但是这里没有用。所以这个推导也适用于多项式。

设 $f^p(x)=(a{x}^n+b{x}^m)\mathrm{m}\mathrm{o}\mathrm{d}p$，则

$$\begin{array}{rl}(a{x}^n+b{x}^m{)}^p& \equiv a^p{x}^{pn}+b^p{x}^{pm}\\ & \equiv a{x}^{pn}+b{x}^{pm}\\ & \equiv f(x^p)(\mathrm{m}\mathrm{o}\mathrm{d}p)\end{array}$$

设 $s=\lfloor \frac{n}{p}\rfloor ,t=\lfloor \frac{m}{p}\rfloor ,q=n\mathrm{m}\mathrm{o}\mathrm{d}p,w=m\mathrm{m}\mathrm{o}\mathrm{d}p$，则有 $n=sp+q,m=tp+w$。

$\begin{array}{rl}(1+x{)}^n& \equiv ((1+x{)}^p{)}^s(1+x{)}^q\\ & \equiv (1+x^p{)}^s+(1+x{)}^q\\ & \equiv \sum _{i=1}^s\left(\genfrac{}{}{0ex}{}{s}{i}\right)x^{pi}\times \sum _{i=1}^q\left(\genfrac{}{}{0ex}{}{q}{j}\right)x^j(\mathrm{m}\mathrm{o}\mathrm{d}p)\end{array}$

又 $(1+x{)}^n=\sum _{i=1}^{sp+q}(\genfrac{}{}{0ex}{}{sp+q}{i})x^i$

$\therefore \sum _{i=1}^{sp+q}\left(\genfrac{}{}{0ex}{}{sp+q}{i}\right)x^i\equiv \sum _{i=1}^s\left(\genfrac{}{}{0ex}{}{s}{i}\right)x^{pi}\times \sum _{i=1}^q\left(\genfrac{}{}{0ex}{}{q}{j}\right)x^j(\mathrm{m}\mathrm{o}\mathrm{d}p)$

可以发现，当 $m\le n$ 时，两边都有 $x^{tp+m}$ 项。对比两边系数：

$$\begin{array}{rl}\left(\genfrac{}{}{0ex}{}{sp+q}{tp+w}\right)x^{tp+w}& \equiv \left(\genfrac{}{}{0ex}{}{s}{t}\right)x^{tp}\times \left(\genfrac{}{}{0ex}{}{q}{w}\right)x^w\\ \left(\genfrac{}{}{0ex}{}{sp+q}{tp+w}\right)& \equiv \left(\genfrac{}{}{0ex}{}{s}{t}\right)\times \left(\genfrac{}{}{0ex}{}{q}{w}\right)\\ \left(\genfrac{}{}{0ex}{}{n}{m}\right)& \equiv \left(\genfrac{}{}{0ex}{}{\lfloor \frac{n}{p}\rfloor }{\lfloor \frac{m}{p}\rfloor }\right)\left(\genfrac{}{}{0ex}{}{n\mathrm{m}\mathrm{o}\mathrm{d}p}{m\mathrm{m}\mathrm{o}\mathrm{d}p}\right)(\mathrm{m}\mathrm{o}\mathrm{d}p)\end{array}$$

得证。

const int N = 1e5 + 10;
long long A[N];//预处理出阶乘
long long qp(long long a, long long b, long long p);//快速幂
long long C(long long n, long long m, long long p){
    if(m > n) return 0;
    return (A[n] * qp(A[m], p-2, p) % p) * 
           qp(A[n-m], p-2, p) % p;
}
long long Lucas(long long n, long long m, long long p){
    if(!m) return 1;
    return C(n%p, m%p, p) * Lucas(n/p, m/p, p) % p;
}
long long solve(long long n, long long m, long long p){
    A[0] = 1;
    for(int i = 1; i <= n; ++ i) A[i] = A[i-1] * i % p;
    return Lucas(n, m, p);
} 

例题

Luogu P2480 [SDOI2010] 古代猪文（基础数论全家桶）

求 $g^{{\sum }_{d|n}{C}_n^d}\mathrm{m}\mathrm{o}\mathrm{d}999911659$。

$1\le n,g\le 10^9$。

• 因为 $999911659\in \mathbb{P}$。
• 所以由欧拉定理可知，$g^{999911658}\equiv 1(\mathrm{m}\mathrm{o}\mathrm{d}999911659)$，注意这里要有个特判：当 $g=999911659$ 时直接输出 $0$ 结束程序。
• 所以原式 $=g^{{\sum }_{d|n}{C}_n^d\mathrm{m}\mathrm{o}\mathrm{d}999911658}\mathrm{m}\mathrm{o}\mathrm{d}999911659$。
• $999911658=2\ast 3\ast 4679\ast 35617$。
• 令 $t={\sum }_{d|n}{C}_n^d\mathrm{m}\mathrm{o}\mathrm{d}999911658$。
• 则 $\{\begin{array}{l}t\equiv {\sum }_{d|n}{C}_n^d(\mathrm{m}\mathrm{o}\mathrm{d}2)\\ t\equiv {\sum }_{d|n}{C}_n^d(\mathrm{m}\mathrm{o}\mathrm{d}3)\\ t\equiv {\sum }_{d|n}{C}_n^d(\mathrm{m}\mathrm{o}\mathrm{d}4679)\\ t\equiv {\sum }_{d|n}{C}_n^d(\mathrm{m}\mathrm{o}\mathrm{d}35617)\end{array}$。
• 每一项用 Lucas，然后用 CRT 求 $t$。

typedef long long ll;
const int N = 35620, P = 999911658;
ll A[N], x[4], y[4] = {2, 3, 4679, 35617};
ll qp(ll a, ll b, ll p);
ll C(ll n, ll m, ll p);
ll Lucas(ll n, ll m, ll p);
ll CRT(){
    ll ans = 0;
    for(int k = 0; k <= 3; ++ k)
        ans = (ans + x[k] * (P/y[k]) % P * qp(P/y[k], y[k]-2, y[k])) % P;
    return ans;
}
ll solve(ll n, ll g){
    if(g == P + 1) return 0;
    for(int k = 0; k <= 3; ++ k){
        A[0] = 1;
        for(ll i = 1; i <= y[k]; ++ i) A[i] = A[i-1] * i % y[k];
        for(ll i = 1; i * i <= n; ++ i) if(n%i == 0){
            x[k] = (x[k] + Lucas(n, i, y[k])) % y[k];
            if(i * i != n) x[k] = (x[k] + Lucas(n, n/i, y[k])) % y[k];
        }
    }
    return qp(g, CRT(), P+1);
}

扩展卢卡斯定理 / $\text{exLucas}$

问题如下：

给定整数 $n,m,p$，求 $\left(\genfrac{}{}{0ex}{}{n}{m}\right)\mathrm{m}\mathrm{o}\mathrm{d}p$，不保证 $p\in \mathbb{P}$。

$1\le n,m\le 10^{18},2\le p\le 10^6$。

其实和 Lucas 没什么关系。以下是解决步骤：

解法

Step 1. 分解 $p$

对于 $p$ 进行质因数分解：$p=\prod _{i=1}^r{p}_i^{k_i}$。将原式化成若干同余方程：

$\{\begin{array}{l}\left(\genfrac{}{}{0ex}{}{n}{m}\right)\equiv a_1(\mathrm{m}\mathrm{o}\mathrm{d}{p}_1^{k_1})\\ \left(\genfrac{}{}{0ex}{}{n}{m}\right)\equiv a_2(\mathrm{m}\mathrm{o}\mathrm{d}{p}_2^{k_2})\\ ...\\ \left(\genfrac{}{}{0ex}{}{n}{m}\right)\equiv a_i(\mathrm{m}\mathrm{o}\mathrm{d}{p}_i^{k_i})\end{array}$

因为 $p_i^{k_i}$ 两两互素，最后使用 CRT 求解即可。

Step 2. 求解 $a_i$

$a_i=\left(\genfrac{}{}{0ex}{}{n}{m}\right)\mathrm{m}\mathrm{o}\mathrm{d}{p}_i^{k_i}=\frac{n!}{m!(n-m)!}\mathrm{m}\mathrm{o}\mathrm{d}{p}_i^{k_i}$，但是 $p_i^{k_i}$ 仍然不是质数，不能求出分母的逆元。所以将分子分母的所有 $p_i$ 提出来：

$a_i=\frac{\frac{n!}{p_i^{k_1}}}{\frac{m!}{p_i^{k_2}}\times \frac{(n-m)!}{p_i^{k_3}}}\times p_i^{k1-k2-k3}$，分母可以求逆元。

Step 3.求解 $\frac{n!}{p^k}$

举个例子：$n=22,p=3,k=2$。

$\begin{array}{rl}n!& =1\ast 2\ast ...\ast 22\\ & =3^7\ast (1\ast 2\ast 3\ast 4\ast 5\ast 6\ast 7)\ast (1\ast 2\ast 4\ast 5\ast 7\ast 8)\ast (10\ast 11\ast 13\ast 14\ast 16\ast 17)\ast (19\ast 20\ast 22)\\ & \equiv 3^7\ast 7!\ast (1\ast 2\ast 4\ast 5\ast 7\ast 8{)}^2\ast (1\ast 2\ast 4)(\mathrm{m}\mathrm{o}\mathrm{d}{p}^k)\end{array}$

分为三个部分

• $3^7$ 为 $p^{\frac{n}{p}}$（会被分母消掉，不用算）。
• $7!$ 递归求解
• 在$\mathrm{m}\mathrm{o}\mathrm{d}{p}^k$ 意义下是循环节，有 $\frac{n}{p^k}$ 个，剩下的那部分暴力求解。

ll calc(ll n, ll p, ll pk){
    if(!n) return 1;
    ll ans = 1;
    for(ll i = 1; i <= pk; ++ i)//每个循环节
        if(i % p) ans = ans * i % pk;
    ans = qp(ans, n/pk, pk);//所有循环节
    for(ll i = 1; i <= n % pk; ++ i)//剩下的
        if(i % p) ans = ans * i % pk;
    return ans * calc(n/p, p, pk) % pk;//递归求解
}

完整代码

typedef long long ll;
const int N = 1e6 + 10;
ll a[N], b[N]; int tot;
ll qp(ll a, ll b, ll p);
void exgcd(ll &x, ll &y, ll a, ll b);
ll inv(ll x, ll p);
ll calc(ll n, ll p, ll pk){
    if(!n) return 1; ll ans = 1;
    for(ll i = 1; i <= pk; ++ i) if(i%p) ans = ans * i % pk;
    ans = qp(ans, n/pk, pk);
    for(ll i = 1; i <= n%pk; ++i) if(i%p) ans = ans * i % pk;
    return ans * calc(n/p, p, pk) % pk;
}
ll C(ll n, ll m, ll p, ll pk){
    if(!n || !m || n==m) return 1; if(n < m) return 0;
    ll nn = calc(n, p, pk), mm = calc(m, p, pk), nm = calc(n-m, p, pk);
    ll cnt = 0, k = n - m;//cnt即上文k1-k2-k3
    while(n) n /= p, cnt += n;
    while(m) m /= p, cnt -= m;
    while(k) k /= p, cnt -= k;
    return nn * inv(mm, pk) % pk * inv(nm, pk) % pk *
           qp(p, cnt, pk) % pk;
}
ll CRT(){
    ll M = 1, ans = 0;
    for(int i = 1; i <= tot; ++ i) M *= b[i];
    for(int i = 1; i <= tot; ++ i)
        ans += a[i] * (M/b[i]) * inv(M/b[i], b[i]);
    return (ans % M + M) % M;
}
ll exLucas(ll n, ll m, ll p){
    for(ll i = 2; i * i <= p && p >= 1; ++ i){
        ll pk = 1;
        while(p%i == 0) p /= i, pk *= i;
        if(pk > 1) a[++tot] = C(n, m, i, pk), b[tot] = pk;
    }
    if(p > 1) a[++tot] = C(n, m, p, p), b[tot] = p;
    return CRT();
}