如果不能正常显示,请查看原文 , 或返回

"3升5升得4升"——倒水问题的万能解法(扩展欧几里得算法)

扩展欧几里得算法及其应用

问题:假设你有一个3升的容器和一个5升的容器(以及充足的水源),如何精确地取出4升水来?(为了下文叙述的方便,我们不妨把3升的容器和5升的容器分别记做容器A和容器B)。这里提供一种解法:

  1. 将A装满(3升),全部倒入B
  2. 再次将A装满(3升),用A中的水将B装满,此时A有1升,B有5升
  3. B中的水全部倒出,将A中的1升水倒入B,此时A没有水,B有一升
  4. 再次将A装满(3升),将A中的水全部倒入B,此时B中恰有4升的水

显然这类问题可以有其他的解决方案。我们可轻易地编出其他类似的问题,比如是否能够用7升的水杯和13升的水杯量出5升的水,再比如能否用9升的水杯和15升的水杯量出10升的水,不胜枚举。三升五升得四升的问题还算直观,稍作思考便可构造解决方案。7升13升得5升,似乎就没那么直观了。而且还有一个问题,首先需要判断是否可行,然后才是给出解决方案。

这样的问题存在一个万能的解法吗?答案是肯定的。注意到,用3升的容器和5升的容器量出4升的水,这一看似复杂的步骤可以简单地概括为:不断地将整杯整杯的A往B里倒,期间只要B被装满就把B倒空。即求解 3xmod5=4,使用 扩展欧几里得算法及其应用 一文的算法我们可轻易解得 x=3。首先根据扩展欧几里得算法,问题有解。x=3 时,对应的解决方案见上文。

接着看 7 升 13 升得 5 升,也即 7xmod13=57,13 互质,首先判断有解,即存在这样一个解决方案。这个方案是怎样的呢?

第一步,首先来看贝祖等式时的情况,也即 7xmod13=1时,此时解得 x=2,也即 2 个 A 倒入 B,A余一升(得1升)

第二步,7xmod13=5,此时 x=(2×5)%13=10,也即对第一步的行为执行五次,如下:

  1. 将装满 7 升水的 A 倒入 B ⇒ A(0升),B(7升)
  2. 将装满7升水的 A 倒入 B,直到 B 加满为止 ⇒ A(1升),B(13升)
  3. 将 B 清空,⇒ A (1升),B(0升)
  4. 将 A 中的一升水倒入 B,⇒ A(0升),B(1升)(1,2,3,4 步为一个单元)
  5. 将 A 加满倒入 B,⇒ A(0升),B(8升)
  6. 再次将 A 加满倒入 B,直到 B 满为止,⇒ A(2升),B(13升)
  7. 将 B 清空,⇒ A(2升),B(0升)
  8. 将 A 中的 2 升水倒入 B,⇒ A(0升),B(2升)
    。。。
    直到 A(5升),B(0升)

我们进一步将问题抽象,用容积分别为 ab 的水杯量出体积为 c 的水,实际上相当于求解方程 axmodb=c。如果 a,b 互质,问题保证有解。如果 c==gcd(a,b) 或者 c==kgcd(a,b),用扩展欧几里得算法便可求解 x,然后得最终的量水方案。如果 c 不能被 gcd(a,b) 整除,方程无解,也即问题无解,比如9升15升的容器得10升的水,10 不能被 gcd(9,15)=3 整除。

def ext_euclid(a, b):
                    # 扩展的欧几里得算法
                    # 用以求解
                    # d = gcd(a, b) = a*x+b*y
    if b == 0:
        return (a, 1, 0)
    d, x, y = ext_euclid(b, a%b)
    return (d, y, x-a//b*y)

def mod_linear_equation(a, b, c):
    d, x, y = ext_euclid(a, b)
    if c % d:
        raise 'no solution'
    return x * c//d % b

if __name__ == '__main__':
    print(mod_linear_equation(3, 5, 4))
                # 3
    print(mod_linear_equation(7, 13, 5))
                # 10
返回