首页 > 代码库 > 求逆元方法总结
求逆元方法总结
在MOD的情况下, (a*b/c ) %MOD 不能直接 / c 来求,需要找到一个数 inv 使得 inv * c % MOD = 1 。 这样 (a*b / c) % MOD = (a * b * inv) % MOD;
性质: 逆元是积性函数 存在 a*b = c ,那么 inv[c] = inv[a] * inv[b] % MOD;
1、 循环找解的方法
long long circleRun(long long n){ for(long long i = 1;i < MOD;i++) if(i * n % MOD == 1) return i; return -1; } long long n; int main(){ while(cin >> n){ cout << n << " 's inv is "<<endl; cout << circleRun(n) << endl; } return 0; }
2、费马小定理的解法 MOD 是质数才能用
利用 a ^ (p-1) % MOD === 1 , 那么它的逆元就是 a ^ (p-2)
#include<stdio.h> #include<string.h> #include<iostream> #include<cmath> using namespace std; const int MOD = 1e9+7; long long quickpow(long long base,long long n){ long long ans = 1; while(n){ if(n%2 == 1) ans = ans * base % MOD; n /= 2; base = base * base % MOD; } return ans; } long long n; int main(){ while(cin >> n){ cout << n << " 's inv is "<<endl; //cout << circleRun(n) << endl; cout << " a ^ (p-2) % MOD "<< endl; cout << quickpow(n, MOD-2) << endl; } return 0; }
3、利用欧几里德扩展来求 ,
欧几里德扩展 是用来解决 ax + by = gcd(a,b)这样的等式。
这时候取 b = MOD, 你可以写成这样 ax = gcd(a,b) - by
推导出 a*x % MOD = gcd(a,b) %MOD
所以只要 gcd(a,b) % MOD === 1时,就可以使用这条来求a的逆元
但用exgcd求得时候,inv可能是负数, 还需要进行如下操作
inv = (inv % MOD + MOD) % MOD;
long long exGcd(long long a, long long b, long long &x0, long long &y0) // a*x0 + b*y0 = gcd(a,b) { if(b==0) { x0 = 1; y0 = 0; return a; } long long r = exGcd(b, a % b, x0, y0); long long t = x0; x0 = y0; y0 = t - a / b * y0; return r; } long long n; int main(){ while(cin >> n){ cout << n << " 's inv is "<<endl; //cout << circleRun(n) << endl; cout << " a ^ (p-2) % MOD "<< endl; cout << quickpow(n, MOD-2) << endl; cout << " ax + by = gcd(a,b) " << endl; long long inv,y0; exGcd(n ,MOD,inv,y0); inv = (inv % MOD + MOD) % MOD; cout << inv << endl; } return 0; }
预处理1-n关于p的逆元:(n < p) , 因为 逆元是积性函数,所以只要 p > n 成立即可,而不需要p必须为素数
假设已经预处理了1-i-1的逆元,j的逆元设为F[j]
令p = x * i –y ( 0 < y < i)
X* i = y (mod p)
X* F[y] * i = y * F[y] = 1(mod p)
所以i的逆元是F[i] = X* F[y]
这样就可以O(n)的时间预处理了。
参考链接
代码实现
const int N = 100005; long long _inv[N]; void pre() { _inv[0] = _inv[1] = 1; for (int i = 2; i < N; i++) { _inv[i] = ((MOD - MOD / i) * _inv[MOD % i]) % MOD; } } long long n; int main(){ pre(); while(cin >> n){ cout << _inv[n] << endl; } return 0; }
4、 利用逆元函数是完全积性函数的性质
求所有质数的逆元即可,预处理复杂度是O(n / logn * logn) = O(n)
这种写法可以用 exgcd 来实现素数的logn 求逆,因为此时 a = p , MOD无论取何值(p除外) , 都有 gcd(p ,mod) = 1,适合通用情况。
之后再采取质因数分解的方法,即可对任意一个 n 以 logn速度求出其逆元
不过。。ACM竞赛如果为了求逆写这么长的代码貌似不太现实。
求逆元方法总结