逆元求法

逆元求法

时间复杂度T(n)：O(log₂n);

空间复杂度S(n)：O(n);

 

去搜罗了一些关于逆元的各种优秀的求取方式，在下面列举出来

前置技能：欧几里得 & 扩展欧几里得

 

/*O(log n) 欧几里得, (最大公约数)*/
long long gcd(long long a, long long b) { return b ? gcd(b, a % b) : a; }

/*O(log n) 扩欧 [ax mod b = 1]*/
long long x, y;
inline void ext_gcd(long long a, long long b, long long &x, long long &y) {
	b ? (ext_gcd(b, a % b, y, x), y -= (a / b) * x) : (x = 1, y = 0);
}

逆元1 ：O(logn)针对单个数字a，求a 在 %p 意义下的逆元（gcd(a,p)要等于1）

inline long long get_inv1(long long a, long long mod) {
	ext_gcd(a, mod, x, y);
	return (x % mod + mod) % mod; 
}

证明：略

逆元2：O(n)预处理阶乘和逆元

/*O(n)阶乘求逆元 + 预处理阶乘*/
inline void get_inv2(const int &n, const int &mod) {

	fac[0] = fac[1] = 1;	
	for (int i = 2; i <= n; ++i) fac[i] = fac[i - 1] * i % mod;
	ext_gcd(fac[n], mod, x, y);
	inv[n] = (x % mod + mod) % mod;
	for (int i = n - 1; i >= 0; --i) inv[i] = inv[i + 1] * (i + 1) % mod;
}

证明：显然针对买最后一个fac[n]需要直接用ext_gcd求取，然后其他的fac[i]每一次其实只需要消除上一个fac[i+ 1]中的i +1的影响就可以了，所以每一次乘上i + 1即可

 

逆元3：O(n)递推求(1 ~ n) 在mod p意义下的逆元

/*O(n)线筛求(1 ~ n)逆元*/
inline void get_inv3(const int &n, const int &mod) {
	inv[1] = 1;
    for (register int i = 2; i <= n; ++i)
    	inv[i] = ((-(mod / i) * inv[mod % i]) % mod + mod) % mod;
}

证明：

   设 令k = p / i, r = p % i, 则p = k * i + r

   则 k * i + r = 0 (mod p)

   两边同时乘上 r^-1 * i^-1,得

   k * r^-1+ i^-1 = 0 (mod p)

   移项得， i^-1= -k * r^-1 (mod p)

   即 i^-1 = -(p / i) *(p mod i)^-1 (mod p)

   那么我们只需要设置边界inv[1]= 1, 就可以通过递推在O(n)时间内求取 (1 ~ n) 在 mod p 意义下的逆元了