求乘法逆元方法总结

简介

若有线性同余方程 $ax\equiv 1\left(modp\right)$，则称 $x$ 为 $a$ 在模 $p$ 意义下的逆元，记作 $a^{-1}$

扩展欧几里得法

Idea

原方程 $ax\equiv 1\left(modp\right)$ 可转化为 $ax+py=1$
然后就可以采用和解线性同余方程相同的方法，通过求解 $ax+py=gcd(a,p)$ 来得到方程的一组解
根据裴蜀定理，我们发现方程有解的条件为 $gcd(a,p)=1$ ，$a$ 和 $p$ 互质
所以得出结论：$a$ 在 模 $p$ 意义下的乘法逆元存在当且仅当 $a$，$p$互质
故题目往往给的模数为一大质数，从而保证了$[1,p-1]$ 中所有整数都有模 $p$ 意义下的逆元

因此，我们只需要求出方程 $ax+py=gcd(a,p)$ 的最小整数解 $x_0$，同时由于 $gcd(a,p)=1$
$x_0$即为 $a^{-1}$

时间复杂度: $O(\log p)$

Code

//b 即为 p
ll exgcd(ll a, ll b, ll &x, ll &y) {
    if (b == 0) {
        x = 1, y = 0;
        return a;
    }
    int d = exgcd(b, a % b, x, y);
    int t = x; x = y; y = t - (a / b) * y;
    return d;
}

快速幂法

Idea

Note ：使用这种方法需满足 $p$ 为质数

费马小定理：若p为质数，且整数 $a$ 满足 $p|a$ ，则有 $a^{b-1}\equiv 1(modp)$

原方程转化为 $a^{p-2}\equiv a^{-1}(modp)$
所以只需要求 $a^{p-2}modp$ 即可，可以用快速幂来求

时间复杂度: $O(\log p)$

Code

//qpow(a,p-2)
inline int qpow(ll a,int n){
	int res = 1;
	a = (a % mod + mod) % mod;
	while(n){
		if(n & 1) res = (res * a) % mod;
		a = (a * a) % mod;
		n >>= 1;
	}
	return res;
}

线性求[1,n]逆元

Note ：使用这种方法需满足 $p$ 为质数

Idea

若要求 $[1,n]$ 每个数的逆元，虽然可以使用上面介绍的两种求单个数逆元的方法
但复杂度来到了 $O(n\log p)$，数据大的时候可能会TLE

故得用线性时间来求

时间复杂度: $O(n)$

Code

inv[1] = 1;
for (int i = 2; i <= n; ++i) inv[i] = (long long)(p - p / i) * inv[p % i] % p;

线性求任意n个数的逆元

Idea

若要求任意n个数 $(a_1,a_2...a_n)$ 的逆元 ($1\le a_i<p$)的逆元， O ( m a x { a i } ) O(max\left\{a_i\right\}) O(max{ai​}) 的复杂度显然不行

就需要使用下面的方法：
先计算出 $n$ 个数的前缀积 $s_i$, 然后用exgcd / qpow 计算 $s_n$ 的逆元，记为 $s{v}_n$

因为 $s{v}_n$ 是 $n$ 个数的积的逆元，所以当我们把它乘上 $a_n$ 时，就会和 $a_n$ 的逆元抵消，于是就得到了 $a_1$ 到 $a_{n-1}$ 的积逆元，记为 $s{v}_{n-1}$

同理我们可以依次计算出所有的 $s{v}_i$，于是 ${a_i}^{-1}$就可以用 $s_{i-1}\ast s{v}_i$ 求得

所以我们就在几乎线性的时间内计算出了 $n$ 个数的逆元

时间复杂度: $O(n+\log p)$

Code

s[0] = 1;
for (int i = 1; i <= n; ++i) s[i] = s[i - 1] * a[i] % p;
sv[n] = qpow(s[n], p - 2);
for (int i = n; i >= 1; --i) sv[i - 1] = sv[i] * a[i] % p;
for (int i = 1; i <= n; ++i) inv[i] = sv[i] * s[i - 1] % p;