【算法讲解】数论——除法取余，费马小定理

notonlysuccess

已于 2023-06-06 16:54:18 修改

阅读量1.4k

点赞数

分类专栏：算法讲解文章标签：算法

于 2023-05-30 20:38:17 首次发布

本文链接：https://blog.csdn.net/notonlysuccess/article/details/130927645

版权

算法讲解专栏收录该内容

6 篇文章

订阅专栏

背景

对于取余运算，有以下一些性质，加减乘都符合分配律
$(a+b)\%p=(a\%p+b\%p)\%p\\ (a\cdot b)\%p=(a\%p\cdot b\%p)\%p\\ (a-b)\%p=(a\%p-b\%p)\%p\\$
但唯独除法不符合分配律
$\frac{a}{b}\%p \neq \frac{a\%p}{b\%p}\%p$
我们可以用乘法分配律很容易计算 $a_1\cdot a_2\cdots a_n\%p$
但当我们计算 $\frac{a_1\cdot a_2\cdots a_n}{b_1 \cdot b_2\cdots b_n}\%p$ 时就，常见的就有组合数 $\binom{n}{r} =\frac{n!}{r!\cdot (n-r)!}$ 的取余

为了解决这个问题，我们需要依赖一个著名定理：费马小定理

费马小定理

当 $p$ 为质数，对于任意的与 $p$ 互质的整数 $a$ ，都有 $a^{p-1}$ 取余 $p$ 等于 $1$

$a^{p-1}\equiv1\mod p, p是质数，a \perp p$
符号说明：
$a\equiv b \mod c$ ：表示对于整数 $a$ , $b$ , $c$ ， $a$ 除以 $c$ 余 $b$
$a\perp b$ ：表示 $a$ , $b$ 互质

解决除法的取余

根据乘法的分配律，可得 $\frac{a}{b}\%p=(a\%p\cdot \frac{1}{b}\%p)\%p$
所以我们需要计算的就是 $\frac{1}{b}\%p$
设 $\frac{1}{b}\equiv b'\%p$ ，我们称 $b^{'}$ 为 $b$ 的逆元
既 $b\cdot b' \equiv1\mod p$
根据费马小定理 $b^{p-1}\equiv1\mod p$
所以 $b^{p-1}\equiv b\cdot b'\mod p$
两边去同类项 $b$ 可得 $\equiv b^{p-2} \mod p$
所以 $\textcolor{Red}{\frac{1}{b}\equiv b^{p-2} \mod p}$
而 $b^{p-2} \mod p$ 可以通过快速幂计算得到

代码

计算组合数 $\binom{n}{r}$

int mod = 998244353;
int qpow(int x, int n) { // 快速幂，求 x 的 n 次方
  int res = 1;
  for (; n; n >>= 1, x = 1LL * x * x % mod) 
    if (n & 1) 
      res = 1LL * res * x % mod;
  return res;
}
int binom(int n, int r) {
  int res = 1;
  for (int i = 1; i <= n; i ++) {
    res = 1LL * res * i % mod;
  }
  for (int i = 1; i <= r; i ++) {
    res = 1LL * res * qpow(i, mod - 2) % mod;
  }
  for (int i = 1; i <= (n - r); i ++) {
    res = 1LL * res * qpow(i, mod - 2) % mod;
  }
  return res;
}