如何用纯 Python 实现“将列表中每个位置替换为其余元素乘积”的高效函数

本文介绍一种不依赖 numpy 的简洁、健壮方法,将输入列表的每个元素替换为除它自身外所有其他元素的乘积,逐位置生成新列表。

在算法题和实际数据处理中,一个经典变体是:对给定列表 arr,构造一个新列表 result,使得 result[i] 等于 arr 中除 arr[i] 外所有元素的乘积。例如:

arr = [69, 78, 62, 73, 23]
# result[0] = 78 * 62 * 73 * 23 = 8119644
# result[1] = 69 * 62 * 73 * 23 = 7182762
# ...

预期输出为:

[8119644, 7182762, 9036378, 7674732, 24358932]

⚠️ 注意:原始代码存在多个关键问题:

  • results = arr 是浅拷贝,修改 results[i] 会污染原列表;
  • 使用 np.prod() 和 / 在含零时会出错(除零或无穷);
  • 循环中反复复用同一列表对象,未真正“每次生成一个独立新列表”。

下面提供 纯 Python、无第三方库、鲁棒支持零值 的实现方案:

✅ 推荐解法:两次遍历(O(n) 时间 + O(1) 额外空间)

def product_except_self(nums):
    """
    返回新列表,其中 result[i] = 所有 nums[j] (j != i) 的乘积
    支持零值,时间复杂度 O(n),空间复杂度 O(1)(不计输出)
    """
    n = len(nums)
    if n == 0:
        return []

    # 第一步:计算每个位置左侧所有元素乘积
    result = [1] * n
    for i in range(1, n):
        result[i] = result[i-1] * nums[i-1]

    # 第二步:从右向左遍历,用变量维护右侧乘积
    right_product = 1
    for i in range(n-1, -1, -1):
        result[i] *= right_product
        right_product *= nums[i]

    return result

# 测试
print([69, 78, 62, 73, 23])
print(product_except_self([69, 78, 62, 73, 23]))
# 输出: [8119644, 7182762, 9036378, 7674732, 24358932]

? 为什么这个解法更优?

  • 无除法 → 完全规避零值导致的 ZeroDivisionError 或浮点异常;
  • 不修改原列表 → 每次返回全新列表,符合函数式设计原则;
  • 空间高效 → 仅用常数额外变量(right_product),输出数组不算额外空间;
  • 可读性强 → 逻辑清晰分为“左积”与“右积”两阶段。

❗ 常见错误规避提示

  • ✖ 错误:results = arr → 应改用 results = arr.copy() 或 [x for x in arr];
  • ✖ 错误:product / arr[i] → 遇到 arr[i] == 0 时崩溃,且浮点精度可能丢失;
  • ✖ 错误:在循环内重复使用同一列表对象 → 必须为每轮生成独立结果(本题虽只要求单个输出列表,但语义上是“每个位置对应一个独立变换”)。

✅ 扩展:若需返回全部“单位置替换”矩阵(如原 NumPy 示例)

只需稍作封装:

def all_replacement_lists(nums):
    """返回列表:每个元素是将 nums 中对应位置替换为其余乘积后的新列表"""
    prod_all = product_except_self(nums)  # 获取目标乘积数组
    result = []
    for i in range(len(nums)):
        new_list = nums.copy()
        new_list[i] = prod_all[i]
        result.append(new_list)
    return result

# 示例
nums = [69, 78, 62, 73, 23]
for lst in all_replacement_lists(nums):
    print(lst)

该实现完全脱离 NumPy,稳定、高效、易测试,适合嵌入生产环境或算法面试场景。