算法：逆元

最新推荐文章于 2021-10-06 17:24:21 发布

FantasticGold

最新推荐文章于 2021-10-06 17:24:21 发布

阅读量258

点赞数

分类专栏：算法

本文链接：https://blog.csdn.net/Fast_G/article/details/82861444

版权

算法专栏收录该内容

15 篇文章 0 订阅

订阅专栏

逆元

定义

一个数 $a$ 的倒数 $a^{-1}$ 满足 $\times a^{-1} = 1$ 。取 $a$ 的逆元 $\mod b)^{-1}$ 满足 $\mod b) \times (a \mod b)^{-1} \mod b = 1$ 。其中 $b$ 为质数， $a$ 的逆元 $\mod b)^{-1} = a^{b-2} \mod b$ 。

证明

由费马小定理得 $a^{b-1} \mod b = 1$ ，即 $(a^{b-2} \mod b) \times (a \mod b) \mod b = 1$ ，故 $\mod b)^{-1} = a^{b-2} \mod b$ 。
费马小定理简单证明如下：取 $\{1,2, \dots ,b-1\}$ ，则 $\mod b = \{1,2, \dots ,b-1\}$ ；取 $\times A = \{a,2a, \dots, (b-1)a\}$ ，则 $\times A \mod b = \{1,2, \dots ,b-1\}$ 仍然成立(此处可见下面证明)。取 $A$ 中所有元素相乘得 $S$ ，取 $\times A$ 中所有元素相乘得 $a^{b-1} \times S$ ，由 $\left\{\begin{array}{cc} S \mod b = (b-1)! \mod b\\ a^{b-1} \times S \mod b = (b-1)! \mod b \end{array}\right.$ 得 $a^{b-1} \mod b = 1$ 。
下面用反证法证明 $\times A \mod b = \{1,2, \dots ,b-1\}$ 。对于任意 $\in a \times A, x \lt y$ ，假设 $\mod b = y \mod b$ ，那么 $\mod b = 0$ ，由 $\left\{\begin{array}{cc} y \mod a = 0\\ x \mod a = 0 \end{array}\right.$ 得 $\mod a = 0$ ，由 $b$ 为质数得 $\times b$ ，故 $\mod (a \times b) = 0$ ，因此 $\ge a \times b$ ，与 $\in a \times A$ 矛盾，由此可得 $\mod b \ne y \mod b$ 。因此 $\times A \mod b$ 有 $(b - 1)$ 个互不相同的元素，即 $\times A \mod b = \{1,2, \dots ,b-1\}$ 。

思路

利用快速幂求 $a$ 的逆元 $a^{b-2} \mod b$ 。

时间复杂度

$O(\log n)$

模板

#include "FP.h"

typedef long long LL;
const LL MOD = 1e9+7;  // the divisor of answer

/**
  * @param a: the number a
  * @return: the inverse of a
  * @other: FP is Fast Power 
  */
LL INV(LL a) {
  return FP(a, MOD-2);
}

应用1

求组合数 $C_n^m \mod b$ 。由 $C_n^m = \frac {n!}{m!(n-m)!}$ 得 $C_n^{m} \mod b = \frac {n!}{m!(n-m)!} \mod b = (n! \mod b) \times (m! \mod b)^{-1}\times((n-m)! \mod b)^{-1}\mod b$ 。先预处理 $1!,2!,\dots,n!$ ，时间复杂度为 $O (n)$ ，再直接求 $C_n^m \mod b$ ，时间复杂度为 $O (1)$ 。

模板1

#include "INV.h"

const LL N = 1e6+10;  // the maximum number

LL fac[N];  // the factorial of numbers
LL inv[N];  // the inverse of factorial

/**
  * @other: initialize fac and inv
  */
void init() {
  inv[0] = fac[0] = 1;  // 0! = 1, INV(1) = 1
  for (int i = 1; i < N; ++i) {
    fac[i] = fac[i-1] * i % MOD;  // i! = (i-1)! * i
    inv[i] = INV(fac[i]);
  }
}

/**
  * @param n: the number n
  * @param m: the number m
  * @return: C(n, m)
  */
LL C(LL n, LL m) {
  return fac[n] * inv[m] % MOD * inv[n-m] % MOD;
}

应用2

已知 $p, q, b$ ，求 $a$ 满足 $\times p \mod b = q \mod b$ 。由 $\times p \mod b) \times (p \mod b)^{-1} = (q \mod b) \times (p \mod b)^{-1}$ 得 $\mod b) \times (p \mod b)^{-1}$ 。

模板2

#include "INV.h"

/**
  * @param p: the number p
  * @param q: the number q
  * @return: the number a
  */
LL cal(LL p, LL q) {
  return q * INV(p) % MOD;
}