乘法逆元学习笔记

一、乘法逆元

如线性同余方程 $ax\equiv 1(\mathrm{m}\mathrm{o}\mathrm{d}b)$ 的解 $x$ 称作 $a$ 在模 $b$ 下的逆元，记作 $x=a^{-1}$。

二、乘法逆元存在的条件

根据线性同余方程组解存在的条件，当且仅当 $\gcd (a,b)=1$ 时，$a$ 在模 $b$ 下存在逆元。

三、求解逆元

1. 扩展欧几里得算法

使用扩展欧几里得算法可以在 $O(\log b)$ 的时间复杂度下求出单个数的逆元。
求解线性同余方程 $ax\equiv 1(\mathrm{m}\mathrm{o}\mathrm{d}b)$ 即可。

void exgcd(long long a, long long b, long long &x, long long &y){
	if (!b){
		x = 1, y = 0;
		return ;
	}
	exgcd(b, a % b, y, x);
	y -= (a / b) * x;
}
long long get_inv(long long a, long long p){
	long long x, y;
	exgcd(a, p, x, y);
	if (x < 0) x += p;
	return x;
}

2. 快速幂

使用快速幂可以在 $O(\log n)$ 的时间复杂度下求出单个数的逆元（模数应为质数）。
当 $b$ 为质数时，由费马小定理得 $a^{b-1}\equiv 1(\mathrm{m}\mathrm{o}\mathrm{d}b)$ ，因此 $a\times a^{b-2}\equiv 1(\mathrm{m}\mathrm{o}\mathrm{d}b)$ ，即 $a$ 的逆元为 $a^{b-2}$ 。代码如下：

long long qPow(long long a, long long b){
	long long ret = 1;
	for (; b; b >>= 1){
		if (b & 1) ret = (ret * a) % p;
		a = (a * a) % p;
	}
	return ret;
}

3. 线性求逆元

使用线性求逆元的方法可以在 $O(b)$ 的时间复杂度下求出 $1$ ~ $b-1$ 所有数在模 $b$ 下的逆元。
从 $1$ 到 $b-1$ 递推。
① $1$ 的逆元为 $1$
② 设当前数为 $i$ ，且 $b=ki+r(r<i)$ ，则 $b\mathrm{m}\mathrm{o}\mathrm{d}i=r$ 。
在模 $b$ 意义下：
$$ki+r\equiv 0(\mathrm{m}\mathrm{o}\mathrm{d}b)$$

两边同时乘以 $i^{-1}{r}^{-1}$ 得：
$$k{r}^{-1}+i^{-1}\equiv 0(\mathrm{m}\mathrm{o}\mathrm{d}b)$$

移项整理：
$$i^{-1}\equiv -k{r}^{-1}\equiv -\lfloor \frac{b}{i}\rfloor (b\mathrm{m}\mathrm{o}\mathrm{d}i{)}^{-1}(\mathrm{m}\mathrm{o}\mathrm{d}b)$$

由于 $b\mathrm{m}\mathrm{o}\mathrm{d}i<i$ ，因此已知，可以 $O(1)$ 递推。代码如下（Luogu 3811 【模板】 乘法逆元）：

#include<bits/stdc++.h>
using namespace std;
const int maxn = 3e6 + 5;
int n, p;
int inv[maxn];
int main(){
	ios::sync_with_stdio(false);
	cin.tie(0);
	cin >> n >> p;
	inv[1] = 1;
	for (int i = 2; i <= n; i ++) inv[i] = (p - (p / i)) * inv[p % i] % p;
	for (int i = 1; i <= n; i ++) cout << inv[i] << '\n';
	return 0;
}

4. 线性求逆元（任意）

接下来介绍一种可以在几乎线性的时间内求出给出数列 { a n } \{a_n\} {an​} 中所有数在模 $b$ 意义下的逆元的方法。
首先计算 { a i } \{a_i\} {ai​} 的前缀积，记为 { s u m n } \{sum_n\} {sumn​} ，即 $su{m}_i={\Pi }_{j=1}^i{a}_j$ 。其次计算 $su{m}_n$ 的逆元，记为 $inv{s}_n$ 。将 $inv{s}_n$ 倒推求出每一个前缀积的逆，记作 { i n v s n } \{invs_n\} {invsn​} ，递推公式为：
$$inv{s}_{i-1}=inv{s}_i\ast a_i$$

有了两个数组，就可以据此得出所有数的逆元 { i n v i } \{inv_i\} {invi​} ：
$$in{v}_i=inv{s}_{i-1}\times su{m}_i$$

代码如下（Luogu 5431 【模板】 乘法逆元2）：

/*
 提示：题目卡常，需要使用快读
*/
#include<bits/stdc++.h>
using namespace std;
const int maxn = 5e6 + 5;

long long n, p, k;
long long a[maxn], sum[maxn], invs[maxn];

template <typename T>
inline void read(T &x){
    char c;
    x = 0;
    int fu = 1;
    c = getchar();
    while(c > 57 || c < 48){
        if(c == 45) fu = -1;
        c = getchar();
    }
    while(c <= 57 && c >= 48){
        x = (x << 3) + (x << 1) + c - 48;
        c = getchar();
    }
    x *= fu;
}

long long qPow(long long a, long long b){
	long long ret = 1;
	for (; b; b >>= 1){
		if (b & 1) ret = (ret * a) % p;
		a = (a * a) % p;
	}
	return ret;
}

int main(){
	read(n), read(p), read(k);
	for (int i = 1; i <= n; i ++) read(a[i]);
	sum[0] = 1;
	for (int i = 1; i <= n; i ++) sum[i] = sum[i - 1] * a[i] % p;
	invs[n] = qPow(sum[n], p - 2);
	for (int i = n - 1; i >= 1; i --) invs[i] = invs[i + 1] * a[i + 1] % p;
	long long mulk = 1, ans = 0;
	for (int i = 1; i <= n; i ++){
		mulk = mulk * k % p;
		ans = (ans + mulk * (invs[i] * sum[i - 1] % p) % p) % p;
	}
	printf("%lld\n", ans);
	return 0;
}