从辗转相除法到求逆元，数论算法初体验

本文始发于个人公众号：TechFlow，原创不易，求个关注

今天是算法和数据结构专题的第22篇文章，我们一起来聊聊辗转相除法。

辗转相除法又名欧几里得算法，是求最大公约数的一种算法，英文缩写是gcd。所以如果你在大牛的代码或者是书上看到gcd，要注意，这不是某某党，而是指的辗转相除法。

在介绍这个算法之前，我们先来看下最大公约数问题。

暴力解法

这个问题应该很明确了，我们之前数学课上都有讲过。给我们纸笔让我们求都没有问题，分解因数找下共同的部分，很快就算出来了。但是用代码实现怎么做呢？

用代码实现的话，首先排除分解因数的方法。因为分解因数复杂度太高了，也很容易想明白，既然要分解因数，那么首先需要获得一定量的质数吧。有了质数之后还要遍历质数，将整数一点一点分解，显然很麻烦，还不如直接暴力了。暴力解法并不复杂，我们直接从1开始遍历，记录下来同时能够整除这两个数的最大数即可。我们暴力的范围也不大，从1到n。

很容易写出代码：

def gcd(a, b):
    ret = 0
    for i in range(min(a, b)):
        if a % i == 0 and b % i == 0:
            ret = i
    return ret

这个很简单，也许你可能还会想出一些优化，比如说首先判断一下a和b之间是否有倍数关系，如果有的话直接就可以得到结果了。再比如说我们i的遍历范围其实可以不用到min(a, b)，如果a和b没有倍数关系的话min(a, b) / 2就可以了。这些都是没有问题的，但是即使加上了这些优化依然改变不了这是一个O(n)算法的本质。

比如说a是1e9，b是1e9-1，毫无疑问这样的做法会超时。

辗转相除法

接下来就轮到正主——辗转相除法出场了，这个算法在《九章算术》当中曾经出现过，叫做更相减损术。不管叫什么，原理都是一样的，它的最核心本质是下面这个式子：

$$gcd(a,b)=gcd(b,r),a=bq+r$$

这个式子就是著名的欧几里得定理，这里的r可以看成是a对b取余之后的结果，也就是说a和b的最大公约数等于b和r的最大公约数。这样我们就把a和b的gcd转移成了b和r，然后我们可以继续转移，直到这两个数之间存在倍数关系的时候就找到了答案。

在我们写代码之前，我们先来看一下这个定理的证明。

我们假设u同时整除a和b，显然这样的u一定存在，因为u至少可以是1，所以：

$$\begin{array}{r}a=su,b=tu\\ r=a-bq=su-tuq=(s-tq)u\end{array}$$

所以可以得到u也整除r，同样我们可以证明能够整除b和r的整数也可以整除a。我们假设v可以同时整除b和r：

b = s v , r = t v a = b q + r = s v q + t v = v ( s q + t ) \begin{aligned} b = sv, r = tv\\ a = bq + r = svq + tv = v(sq + t) \end{aligned}