如何简化Python中计算三个整数GCD的动态算法代码？

现有一段计算三个随机整数最大公约数的Python代码，其逻辑为：通过前3个if语句处理两个输入数为0的情况，后续if/elif调整参数顺序；基于gcd(a,b,c)=gcd(gcd(a,b),gcd(a,c))的原理，用嵌套while循环计算GCD。现需优化代码，使其更简洁、更具动态性。

原始代码 #

def GCD3(A, B, C):
    if A == B == 0://
        return C
    if A == C == 0:
        return B
    if B == C == 0://
        return A
    if B == 0:
        B = A
        A = 0
    elif C == 0:
        C = A
        A = 0
    gcd1 = A
    gcd2 = A
    while True:
        r = gcd1 % B
        gcd1 //= B
        gcd1 = B
        B = r
        if B == 0:
            while True:
                r = gcd2 % C
                gcd2 //= C
                gcd2 = C
                C = r
                if C == 0:
                    while True:
                        r = gcd1 % gcd2
                        gcd1 //= gcd2
                        gcd1 = gcd2
                        gcd2 = r
                        if gcd2 == 0:
                            return abs(gcd1)


print(GCD3(2, 0, 0))
print(GCD3(115, 5, 0))
print(GCD3(49, 27, 31))
print(GCD3(56, 12, 48))
print(GCD3(8, 56, 4))

原始代码的问题 #

• 代码冗余：嵌套的while循环重复了欧几里得算法逻辑，维护成本高
• 扩展性差：仅支持固定3个整数的GCD计算，无法直接扩展到更多数
• 边界处理繁琐：手动处理两数为0的情况，逻辑复杂且未覆盖三数全0的异常场景
• 可读性低：嵌套循环和参数调整逻辑混乱，难以快速理解

优化方案 #

思路 #

1. 封装通用的两数GCD函数，复用欧几里得算法逻辑
2. 利用GCD的结合律（gcd(a,b,c) = gcd(gcd(a,b), c)），迭代计算任意数量整数的GCD
3. 自动处理含0的场景（gcd(x,0)=abs(x)），明确处理全0的异常情况

优化后代码 #

def gcd_two(a, b):
    # 先取绝对值，支持负数输入
    a, b = abs(a), abs(b)
    # 欧几里得算法核心逻辑
    while b != 0:
        a, b = b, a % b
    return a

from functools import reduce

def gcd_multiple(*args):
    # 处理全0的异常情况：所有数为0时无有效GCD
    if all(num == 0 for num in args):
        raise ValueError("所有输入数均为0，无最大公约数定义")
    # 过滤掉所有0，因为gcd(x,0)=x，不影响最终结果
    non_zero_nums = [num for num in args if num != 0]
    # 用reduce迭代计算所有非零数的GCD
    return reduce(gcd_two, non_zero_nums)

# 测试用例
print(gcd_multiple(2, 0, 0))       # 输出：2
print(gcd_multiple(115, 5, 0))     # 输出：5
print(gcd_multiple(49, 27, 31))    # 输出：1
print(gcd_multiple(56, 12, 48))    # 输出：4
print(gcd_multiple(8, 56, 4))      # 输出：4

优化点说明 #

• 代码复用：将两数GCD的逻辑单独封装，避免重复编写循环代码
• 动态扩展性：通过*args支持任意数量的整数输入，不仅限于3个
• 健壮性提升：明确处理全0的异常场景，自动过滤0值，简化边界逻辑
• 可读性增强：逻辑分层清晰，借助reduce函数直观体现GCD的结合律
• 兼容性好：支持负数输入，通过取绝对值统一处理

内容的提问来源于stack exchange，提问作者user22169233