Python integer_factorization函数返回嵌套结构问题求助

你的函数出现嵌套输出的核心原因是递归返回结构不统一：有时返回单个素数（如n为素数时return n），有时返回元组+列表（如return integer_factorization(n//i), factors_list），递归叠加后就产生了多层嵌套。下面是具体修正方案：

第一步：修复素数检查函数的bug #

原is_prime会错误将2判定为非素数（因2的末位是2，触发偶数判断返回False），这会影响分解结果，先修正该问题：

import math 
def is_prime(n):
    assert n >= 2
    # 单独处理2、3这两个小素数
    if n == 2 or n == 3:
        return True
    sqrt_n = math.ceil(math.sqrt(n))
    last_digit = n % 10
    # 偶数（除2外）直接返回False
    if last_digit in {0,2,4,6,8}:
        return False
    # 大于5且末位为5的数直接返回False
    if n > 5 and last_digit == 5:
        return False
    # 只检查奇数因子，优化循环效率
    for i in range(3, sqrt_n + 1, 2):
        if n % i == 0:
            return False
    return True

第二步：重构因数分解函数，统一返回结构 #

修改integer_factorization，让它始终返回扁平因数列表，递归时直接合并结果：

def integer_factorization(n):
    # n是素数时，返回包含该素数的列表
    if is_prime(n):
        return [n]
    # 先检查小素数因数
    divisors_list = [2, 3, 5, 7]
    for i in divisors_list:
        if n % i == 0:
            # 将当前因数与n//i的分解结果拼接成列表返回
            return [i] + integer_factorization(n // i)
    # 处理候选列表外的因数（如11、13这类素数的乘积）
    for i in range(11, int(math.sqrt(n)) + 1, 2):
        if n % i == 0:
            return [i] + integer_factorization(n // i)
    # 兜底逻辑：理论上到这里n已是素数
    return [n]

测试验证 #

调用integer_factorization(20)会返回[2, 2, 5]，完全符合你需要的扁平格式；调用integer_factorization(143)会返回[11, 13]，覆盖更多分解场景。

关键修改点说明 #

1. 统一返回类型：无论n是素数还是合数，函数都返回列表，递归时直接拼接，彻底消除嵌套。
2. 简化逻辑：移除原函数中多余的factors_list变量，直接通过列表拼接传递结果。
3. 补充分支覆盖：增加了候选列表外的因数检查，避免无法分解大素数乘积的情况。

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