题目:
给你一个整数数组 nums,返回 数组 answer ,其中 answer[i] 等于 nums 中除 nums[i] 之外其余各元素的乘积 。
题目数据 保证 数组 nums之中任意元素的全部前缀元素和后缀的乘积都在 32 位 整数范围内。
请不要使用除法,且在 O(n) 时间复杂度内完成此题。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/product-of-array-except-self
思路:
简单方法:遍历一遍计算每个数的前缀积和后缀积,使用两个列表prefix和postfix储存每个数的前缀积和后缀积,最后遍历这两个列表计算结果res[i] = prefix[i] * postfix[i + 1]
class Solution:
def productExceptSelf(self, nums: List[int]) -> List[int]:
length = len(nums)
x = y = 1
prefix = [1]
postfix = [1]
for i in range(length):
x *= nums[i]
y *= nums[length - i - 1]
prefix.append(x)
postfix.append(y)
postfix.reverse()
res = []
for i in range(length):
res.append(prefix[i] * postfix[i + 1])
return res
进阶方法:根据题目题目进阶要求,要使用O(1)的额外空间,就不能使用两个列表来储存计算出的前后缀积,这时候就要转变一下思路。
已知前缀级必须由前往后遍历,后缀积必须由后向前遍历,且res[i] = prefix[i] * postfix[i + 1]
令res[i]初始为1,按照循环顺序从res的角度看,
res[i] *= 当前循环的prefix,
res[length - i - 1] *= 当前循环的postfix;
从prefix和postfix角度看,prefix*= nums[i],postfix *= nums[length - i - 1]
则直接遍历一次使用两个变量prefix和postfix即可得出答案res
class Solution:
def productExceptSelf(self, nums: List[int]) -> List[int]:
length = len(nums)
prefix = 1
postfix = 1
res = [1 for i in range(length)]
for i in range(length):
res[i] *= prefix
prefix *= nums[i]
res[length - i - 1] *= postfix
postfix *= nums[length - i - 1]
return res
此时空间效率已为O(1)