本文介绍一种不依赖 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]
⚠️ 注意:原始代码存在多个关键问题:
下面提供 纯 Python、无第三方库、鲁棒支持零值 的
实现方案:
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]只需稍作封装:
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,稳定、高效、易测试,适合嵌入生产环境或算法面试场景。