【欧几里得算法以及扩展欧几里得算法】

欧几里得算法

假设有两个非负整数$a,b$,并且$a,b$都不等于0，那么$a,b$的最大公约数等于其中较小的数和$a,b$相除的余数的最大公约数。

用公式表达为$gcd(a,b)=gcd(b,a(\mathrm{mod}b))$，其中$a>=b$。

证明过程

$$\begin{array}{rl}假设k可以同时整除a,b。& \\ & \because a>=b\\ & \therefore a=kb+r\\ & \therefore k也可以整除r。\\ & \therefore gcd(a,b)=gcd(b,a(\mathrm{mod}b))成立。\end{array}$$

代码实现

def gcd_by_euclidean_algorithm(a: int, b: int) -> int:
    while b != 0:
        a, b = b, a % b
    return a


# expect 1
print(gcd_by_euclidean_algorithm(3, 4))
# expect 4
print(gcd_by_euclidean_algorithm(12, 4))
# expect 6
print(gcd_by_euclidean_algorithm(18, 24))

扩展欧几里得算法

已知正整数$a,b$，求解一组$x,y$，使它们满足贝祖等式$ax+by=gcd(a,b)$。

特别的，当$a\perp b$时，求解的$x$为$a$模$b$的逆元，即$ax\equiv 1(\mathrm{mod}b)$。

扩展欧几里得算法的代码实现

扩展欧几里得算法实质上是求解贝祖等式的解。

def extended_gcd(a: int, b: int) -> int:
    # 当a为0时，贝祖等式的系数x=0,y=1,最大公约数为b
    if a == 0:
        return b, 0, 1
    # 递归求取最大公约数和贝祖系数
    d, x, y = extended_gcd(b % a, a)
    # 更新系数
    return d, y - (b // a) * x, x

根据扩展欧几里得算法求解模逆元

代码实现如下：

def mod_inverse(a, m):
    gcd, x, y = extended_gcd(a, m)
    # 当a,b不互质时，不存在模逆元
    if gcd != 1:
        raise Exception("Not found mod inverse.")
    return x % m

注意： 调用扩展欧几里得算法求得$x$可能是负数，这时需要对$x$模$m$，确保$x$是一个正数。

求证扩展欧几里得算法中系数公式

扩展欧几里得算法的核心是更新贝祖等式系数的公式，这些公式是基于欧几里得算法的递归性质推导出来的。下面是详细的证明过程：

初始等式:$ax+by=gcd(a,b)$，其中$x,y$是贝祖系数。

递归的第二层存在等式：$b(\mathrm{mod}a)\ast x_1+a\ast y_1=gcd(b(\mathrm{mod}a),a)$。

根据欧几里得算法可知：$gcd(a,b)=gcd(b(\mathrm{mod}a),a)=d$，其中$d$表示$a,b$的最大公约数。

又因为：$$b(\mathrm{mod}a)=b-\lfloor \frac{b}{a}\rfloor \ast a$$
将此式子带入第二层的递归的等式中可得：

$$\begin{gathered} (b-\lfloor \frac{b}{a}\rfloor \ast a)\ast x_1+a\ast y_1=d \\ a\ast (y_1-\lfloor \frac{b}{a}\rfloor \ast x_1)+b\ast x_1=d \end{gathered}$$

因此递归时，两次递归之间的系数关系为：

$$\begin{array}{rl}x& =y_1-\lfloor \frac{b}{a}\rfloor \ast x_1\\ y& =x_1\end{array}$$