【数论】求逆元的四种方法

辞树c

已于 2022-12-02 20:51:10 修改

阅读量1.6w

点赞数 29

分类专栏：数论文章标签：算法 c++ 数学

于 2021-10-06 17:24:21 首次发布

本文链接：https://blog.csdn.net/qq_58207591/article/details/120626244

版权

数论专栏收录该内容

6 篇文章 4 订阅

订阅专栏

逆元的定义

给定正整数a，p，如果有 $ax \equiv 1 \: (mod \; p)$ ，且a与p互质，则称x的最小正整数解为a模p的逆元。

一、扩展欧几里得算法

使用条件：基本上通用，不要求p为质数，且效率高，时间复杂度为 $O(log \: n)$ 。

证明过程： $ax \equiv 1 \: (mod \; p)$ 有解的条件是 $gcd(a, p) = 1$ ，即a、p互质，所以根据扩展欧几里得原理，就把问题等价于求解 $ax + py = 1$ ，就可以使用欧几里得算法了。

void extend_gcd(ll a, ll b, ll &x, ll &y){
    if(b == 0){
        x = 1, y = 0;
        return;
    }
    extend_gcd(b, a % b, x, y);
    ll tmp = x;
    x = y;
    y = tmp - (a / b) * y;
}
ll mod_inverse(ll a, ll mod){
    ll x, y;
    extend_gcd(a, mod, x, y);
    return (x % mod + mod) % mod;
}

二、费马小定理

定理内容：如果p是一个质数，而整数a不是p的倍数，则有a^（p-1）≡1（mod p）。

使用条件：要求p为质数，效率也挺高，但由于扩展欧几里得算法通常情况下更优，因此该方法使用情况较少，时间复杂度为 $O(log \: n)$ 。

证明过程：要证明费马小定理，需要两个引理的辅助：

引理1（剩余系定理2）：若a，b，c为任意3个整数，m为正整数，且(m,c) = 1，则当a·c≡b·c(mod m)时，有a≡b(mod m)。

引理2（完全剩余系性质2）：若 $a_{i}$ ( $1 \leqslant i \leqslant n$ )构成模n的完系， $k, m \in Z$ ，(m,n) = 1，则 $k + ma_{i} (1 \leqslant i \leqslant n)$ 也构成模n的完系。

有了这两个引理，我们就可以开始证明了：

对素数p的完全剩余系 $\left \{ 1, 2, 3, ..., p - 1 \right \}$ ，由于(a, p) = 1，根据引理2， $\left \{ a, 2a, 3a, ... , (p - 1)a \right \}$ 也是p的一个完全剩余系，由完全剩余系的性质， $1 \times 2 \times 3 \times ... \times (p - 1) \equiv a \cdot 2a \cdot 3a \cdot ...\cdot (p - 1)a \; (mod \; p)$ ；

即 $(p - 1)! \equiv (p - 1)! \cdot a^{p - 1} \; (mod \; p)$ ；

所以 $a^{p - 1} \equiv 1 \; (mod \; p)$ 得证。

再对其作变形得到 $a \cdot a^{p - 2} \equiv 1 \; (mod \; p)$ ；

令a的逆元为 $a^{-1}$ ，则 $a^{-1} \equiv a^{p - 2} \; (mod \; p)$ 。

ll qmi(ll a, ll b, ll mod){
    ll res = 1;
    while(b){
        if(b & 1) res = res * a % mod;
        a = a * a % mod;
        b >>= 1;
    }
    return res;
}
ll fermat(ll a, ll mod){
    return qmi(a, mod - 2, mod);
}

三、欧拉定理

定理内容：设 $m,n\in N^{+}$ ，且 $gcd(a, m) = 1$ ，则我们有: $a^{\varphi (m)} \equiv 1 \; (mod \; m)$ ，其中 $\varphi (m)$ 称为对模 $m$ 缩系的元素个数。

使用条件：不要求p为质数，这相当于费马小定理求逆元的扩展，先求出欧拉函数，再求逆元，时间复杂度为 $O(sqrt(n))$ 。

证明过程：取模 $m$ 的缩系 $a_1, a_2, ... , a_{\varphi (m)}$ ，则 $aa_1,aa_2,...,aa_{\varphi (m)}$ 也是模 $m$ 的缩系，故有 $\prod_{i = 1}^{\varphi (m)} a_i \equiv \prod_{i = 1}^{\varphi (m)} aa_i \equiv a^{\varphi (m)}\prod_{i = 1}^{\varphi (m)} a_i \; (mod \; m) \Rightarrow a^{\varphi (m)} \equiv 1 \; (mod \; m)$ 得证。

再对其作变形得到 $a \cdot a^{\varphi (m) - 1} \equiv 1 \; (mod \; m)$ ；

令a的逆元为 $a^{-1}$ ，则 $a^{-1} \equiv a^{\varphi (m) - 1} \; (mod \; m)$ 。

ll qmi(ll a, ll b, ll mod){
    ll res = 1;
    while(b){
        if(b & 1) res = res * a % mod;
        a = a * a % mod;
        b >>= 1;
    }
    return res;
}
ll get_phi(ll n){
    ll res = n;
    for(int i = 2; i <= n / i; i++){
        if(n % i == 0){
            res = res / i * (i - 1);
            while(n % i == 0)
                n /= i;
        }
    }
    if(n > 1)
        res = res / n * (n - 1);
    return res;
}
ll mod_inverse(ll a, ll mod){
    if(a % mod != 0)
        return qmi(a, get_phi(mod) - 1, mod);
    else
        return -1;
}

四、递推打表

递推公式： $inv[i] = (mod - mod \; / \; i) * inv[mod \; % \; i] \; % \; mod$ 。

使用条件：要求p为质数，也是一种常用的求逆元的方法，时间复杂度为 $O(n)$ 。

证明过程： $m \; / \; i \; * \; i + m \; % \; i \equiv 0 \; (mod \; m)$

$\Rightarrow$ $- m \; / \; i \; * \; i \equiv m \; % \; i \; (mod \; m)$

$\Rightarrow$ $- m \; / \; i \; * \; inv[m \; % \; i] \equiv inv[i] \; (mod \; m)$

$\Rightarrow$ $inv[i] \equiv (m - m \; / \; i) \; * \; inv[m \; % \; i] \; (mod \; m)$ 证毕。

ll inv[N];   // 逆元表

void mod_inverse(ll n, ll mod){
    inv[0] = inv[1] = 1;
    for(int i = 2; i <= n; i++)
        inv[i] = (mod - mod / i) * inv[mod % i] % mod;
}