【数论算法】扩展欧几里得算法详解+编程求解不定方程

扩展欧几里得算法详解及不定方程求解模板题（洛谷P5656）解题思路

题目链接

https://www.luogu.com.cn/problem/P5656

题目描述

给定不定方程

$$ax+by=c$$

若该方程无整数解，输出 $-1$。

若该方程有整数解，且有正整数解，则输出其正整数解的数量，所有正整数解中 $x$ 的最小值，所有正整数解中 $y$ 的最小值，所有正整数解中 $x$ 的最大值，以及所有正整数解中 $y$ 的最大值。

若方程有整数解，但没有正整数解，你需要输出所有整数解中 $x$ 的最小正整数值， $y$ 的最小正整数值。

正整数解即为 $x,y$ 均为正整数的解，$0$ 不是正整数。

整数解即为 $x,y$ 均为整数的解。

$x$ 的最小正整数值即所有 $x$ 为正整数的整数解中 $x$ 的最小值，$y$ 同理。

数据范围

共有$T$组数据，$1\le T\le 2\times 10^5$，$1\le a,b,c\le 10^9$。

扩展欧几里得算法

本题是扩展欧几里得算法的模板题，下面先介绍扩展欧几里得算法。

代码模板

给定 $a$ 与 $b$，下面的扩展欧几里得算法可以找到一组数 $x$， $y$ 与 $d$，满足 $ax+by=d$ 并且 $d=\gcd (a,b)$。

void ext_gcd(long long a, long long b, long long &d, long long &x, long long &y){
    if (b == 0){
        d = a;
        x = 1; 
        y = 0;
    }else{
        ext_gcd(b, a % b, d, y, x);// (*)
        y -= a / b * x;
    }
}

时间复杂度

上面算法的时间复杂度为 $O(\log (\max (a,b)))$,且最坏情况是计算斐波那契数列的相邻两项。

算法正确性的证明

用数学归纳法可以证明算法的正确性，即算法对于任意合法的自然数 $a$ 与 $b$，均可以找到一组数 $x$， $y$ 与 $d$，满足 $ax+by=d$ 并且 $d=\gcd (a,b)$。

归纳基础

$b=0$ 时， $\gcd (a,0)=a$，取 $x=1,y=0$，有 $ax+by=a\times 1+0\times 0=a=\gcd (a,0)=\gcd (a,b)$ 成立。

归纳步

假设第 $i$ 次递归之后，代码中的 (*) 行已经得到了正确的结果，即返回值 $y^{\prime }$ 和 $x^{\prime }$ 满足：

$$b{y}^{\prime }+(a\mathrm{m}\mathrm{o}\mathrm{d}b)x^{\prime }=\gcd (b,a\mathrm{m}\mathrm{o}\mathrm{d}b)=\gcd (a,b)=d$$

则经过步骤 $y-=\lfloor a/b\rfloor \times x^{\prime }$ 之后，新的数对 $(x,y)$ 满足：

$$ax+by=a{x}^{\prime }+b(y^{\prime }-\lfloor a/b\rfloor x^{\prime })=a{x}^{\prime }+b{y}^{\prime }-b\lfloor a/b\rfloor x^{\prime }$$
$$=(a-b\lfloor a/b\rfloor )x^{\prime }+b{y}^{\prime }=(a\mathrm{m}\mathrm{o}\mathrm{d}b)x^{\prime }+b{y}^{\prime }=d=\gcd (a,b)$$

由数学归纳法得知，任意步骤之后，上面的算法都会返回正确的结果，即都会找到一组数 $x$， $y$ 与 $d$，满足 $ax+by=d$ 并且 $d=\gcd (a,b)$。

算法时间复杂度的证明

我们分几种情况考虑算法的执行。

$b=0$ 时

到达递归出口，执行三条赋值语句，这三条语句的时间复杂度为 $O(1)$。

$b\ne 0$ 且 $a<b$ 时

这种情况需要递归。传入 (*) 行递归调用的前两个参数 $a^{\prime }=b$， $b^{\prime }=a\mathrm{m}\mathrm{o}\mathrm{d}b=a$， 这时有 $a^{\prime }=b>b^{\prime }=a$， 递归后 $a^{\prime }$ 与 $b^{\prime }$ 的关系将变为下一种 $a^{\prime }\ge b^{\prime }$ 的情况。递归调用之外的时间开销为 $O(1)$。

$b\ne 0$ 且 $a\ge b$ 时

这种情况也需要递归。传入 (*) 行递归调用的前两个参数 $a^{\prime }=b$， $b^{\prime }=a\mathrm{m}\mathrm{o}\mathrm{d}b$。由于 $b^{\prime }=a\mathrm{m}\mathrm{o}\mathrm{d}b<b=a^{\prime }$，因此递归后 $a^{\prime }$ 与 $b^{\prime }$ 的关系依然为 $a^{\prime }\ge b^{\prime }$ 的情况。注意到，当 $a\ge b$ 时， $a\mathrm{m}\mathrm{o}\mathrm{d}b\le \lfloor \frac{a}{2}\rfloor$ （证明在下面）。 因此我们有下一次递归调用的前两个参数 $a^{\prime \prime }$ 和 $b^{\prime \prime }$ 满足 $a^{\prime \prime }=b^{\prime }=a\mathrm{m}\mathrm{o}\mathrm{d}b\le \lfloor \frac{a}{2}\rfloor$， 进而有
$$\max (a^{\prime \prime },b^{\prime \prime })=a^{\prime \prime }\le \lfloor \frac{a}{2}\rfloor =\lfloor \frac{\max (a,b)}{2}\rfloor$$

因此在 $a\ge b$ 的情况下，每两次递归调用就可以让递归函数前两个参数 $a$ 与 $b$ 的最大值减少至少一半。而递归调用之外的时间开销为 $O(1)$， 因此算法总的时间复杂度为 $O(\log (\max (a,b)))$。

引理： $\forall a,b\in {\mathbb{Z}}^{+}$，若 $a\ge b$，则有 $a\mathrm{m}\mathrm{o}\mathrm{d}b\le \lfloor \frac{a}{2}\rfloor$

若 $b\le \lfloor \frac{a}{2}\rfloor +1$， 则 $a\mathrm{m}\mathrm{o}\mathrm{d}b\le b-1=\lfloor \frac{a}{2}\rfloor$，引理成立；

若 $b>\lfloor \frac{a}{2}\rfloor +1$， 由于 $a\ge b$，因而有 $\lfloor \frac{a}{b}\rfloor \ge 1$，因此

$$a\mathrm{m}\mathrm{o}\mathrm{d}b=a-\lfloor \frac{a}{b}\rfloor b\le a-b<a-\lfloor \frac{a}{2}\rfloor -1$$
$$=2\lfloor \frac{a}{2}\rfloor +(a\mathrm{m}\mathrm{o}\mathrm{d}2)-\lfloor \frac{a}{2}\rfloor -1=\lfloor \frac{a}{2}\rfloor +(a\mathrm{m}\mathrm{o}\mathrm{d}2)-1\le \lfloor \frac{a}{2}\rfloor$$

因而引理成立。

解题步骤

运用扩展欧几里得算法

对于题目输入的三个数 $a,b,c$，首先可以通过扩展欧几里得算法求出一组数 $x_0,y_0,d$，满足 $a{x}_0+b{y}_0=d$ 且 $d=\gcd (a,b)$。

原方程是否有整数解？

首先检查 $c$ 是不是 $d$ 的倍数，如果不是，则说明无解（因为方程左边 $ax+by$ 一定是 $d$ 的倍数，但右边 $c$ 不是 $d$ 的倍数）。

如何求出原方程的一组特解？

接着将方程 $x_{0}a+y_{0}b=d$ 两边同乘以 $\frac{c}{d}$，得到 $a{x}_0\frac{c}{d}+b{y}_0\frac{c}{d}=d\frac{c}{d}=c$，所以 $(x_0\frac{c}{d},y_0\frac{c}{d})$ 是不定方程 $ax+by=c$ 的一组解。

如何通过特解找到其他的解？

假设有两组解 $(x_1,y_1)$ 与 $(x_2,y_2)$ 使得原不定方程成立，那么有 $a{x}_1+b{y}_1=a{x}_2+b{y}_2=c$，因此 $a(x_1-x_2)=b(y_2-y_1)$，该等式两边同时除以 $d=\gcd (a,b)$，得到：

$$\frac{a}{d}(x_1-x_2)=\frac{b}{d}(y_2-y_1)$$

此时 $\gcd (\frac{a}{d},\frac{b}{d})=1$，因此如果要让等式成立， $x_1-x_2$ 必须要为 $\frac{b}{d}$ 的整数倍。不妨设 $x_1-x_2=t\frac{b}{d}$，那么有 $y_2-y_1=t\frac{a}{d}$。

因此给定一组特解 $(x,y)$，我们可以将不定方程所有可能的解表示为 $(x+t\frac{b}{d},y-t\frac{a}{d})$，其中 $t$ 为任意整数。结合上一小节的分析，我们可以将这一题中的所有可能的解表示为 $(x_0\frac{c}{d}+t\frac{b}{d},y_0\frac{c}{d}-t\frac{a}{d})$，其中 $t$ 为任意整数。

如何找到 $x$ 为尽可能小的正整数的解？

我们已经有了一组初始解 $(x_0\frac{c}{d},y_0\frac{c}{d})$。要找到 $x$ 为尽可能小的正整数的解，实际上就是要找到 $t_{x_{min}}$，使得 $x_0\frac{c}{d}+t_{x_{min}}\frac{b}{d}$ 为尽可能小的正整数 $x_{min}$，即要让 $x_{min}=x_0\frac{c}{d}+t_{x_{min}}\frac{b}{d}$ 为尽可能小的正整数。

注意到 $1\le x_{min}\le \frac{b}{d}$ （若 $x_{min}<1$，则 $x_{min}$ 不为正数；若 $x_{min}>\frac{b}{d}$，则可以让 $x_{min}$ 减小为一个更小的解。因此让 $x_0\frac{c}{d}+t_{x_{min}}\frac{b}{d}=x_{min}$ 两边同时整除以 $\frac{b}{d}$，可以得到 $\lfloor \frac{x_{0}c/d}{b/d}\rfloor +t_{x_{min}}=0$ 或者 $\lfloor \frac{x_{0}c/d}{b/d}\rfloor +t_{x_{min}}=1$，即 $t_{x_{min}}=-\lfloor \frac{x_{0}c/d}{b/d}\rfloor$ 或者 $t_{x_{min}}=1-\lfloor \frac{x_{0}c/d}{b/d}\rfloor$。

因此，我们计算出 $x_0\frac{c}{d}+(-\lfloor \frac{x_{0}c/d}{b/d}\rfloor )\frac{b}{d}$ 以及 $x_0\frac{c}{d}+(1-\lfloor \frac{x_{0}c/d}{b/d}\rfloor )\frac{b}{d}$，看一下哪一个是 $x$ 最小的正整数值。

求出 $x$ 最小的正整数值 $x_{min}$ 之后，我们可以进一步求出当前的 $y=\frac{c-ax}{b}$，将其记为 $y_{max}$。

情况一：原方程不存在正整数解

如果 $y_{max}$ 小于等于 $0$，则说明原不定方程不存在 $x$ 与 $y$ 均为正整数的解（因为要使得方程 $ax+by=c$ 成立，$y$ 变大时需要让 $x$ 变小，但 $x_{min}$ 再变小就不为正整数了，因此 $x_{min}$ 无法变小，$y_{max}$ 也无法变大）。此时可以用类似于上一小节的方法求出 $y_{min}$，然后可以输出 $x_{min}$ 与 $y_{min}$。

情况二：原方程存在正整数解

如果 $y_{max}$ 大于 $0$，则 $(x_{min},y_{max})$ 为原方程的一组正整数解，且 $x_{min}$ 为可以达到的最小整数 $x$ 值，$y_{max}$ 为可以达到的最大整数 $y$ 值。用类似的方法可以求出 $(x_{max},y_{min})$。进而我们可以求出解的数目为 $n=\frac{x_{max}-x_{min}}{b/d}+1$，即 $n=t_{x_{max}}-t_{x_{min}}+1$。然后将 $n,x_{min},y_{min},x_{max},y_{max}$ 均输出即可。

代码

模板题代码就不贴了，理解了上一节解题步骤后按照解题步骤调用模板并实现算术运算与条件判断等语句即可。

相关思考题

1. 如果要解含有三个未知数的不定方程 $ax+by+cz=e$，该怎样处理呢？其中 $x,y,z$ 为未知数， $a,b,c,e$ 为输入的常数。
   • 基本思路是先引入一个新的未知数 $w$，解未知数为 $x,w$ 的不定方程 $ax+gcd(b,c)w=e$，然后再解不定方程 $by+cz=gcd(b,c)$。
2. 如果要解含有 $n$ 个未知数的不定方程，该怎样处理呢？
   • 基本思路类似上一题，引入一个新的未知数转化为解含有 $n-1$ 个未知数的不定方程，然后可以递归处理。