给你一个整数数组 nums,返回 数组 answer ,其中 answer[i] 等于 nums 中除 nums[i] 之外其余各元素的乘积 。
题目数据 保证 数组 nums之中任意元素的全部前缀元素和后缀的乘积都在 32 位 整数范围内。
请 不要使用除法,且在 O(n) 时间复杂度内完成此题。
示例 1:
输入: nums = [1,2,3,4]
输出: [24,12,8,6]
示例 2:
输入: nums = [-1,1,0,-3,3]
输出: [0,0,9,0,0]
ans1
class Solution:
def productExceptSelf(self, nums: List[int]) -> List[int]:
n = len(nums)
ans = [1] * n
prefix = [1] * n
suffix = [1] * n
for i in range(1, n):
prefix[i] = prefix[i - 1] * nums[i - 1]
for i in range(n-2,-1,-1):
suffix[i] = suffix[i + 1] * nums[i + 1]
for i in range(n):
ans[i] = prefix[i] * suffix[i]
return ans
ans2
class Solution:
def productExceptSelf(self, nums: List[int]) -> List[int]:
n = len(nums)
ans = [1] * n
# 计算前缀积并直接存入 ans
for i in range(1, n):
ans[i] = ans[i - 1] * nums[i - 1]
# 动态计算后缀积并与前缀积相乘
suffix = 1
for i in range(n-1, -1, -1):
ans[i] *= suffix
suffix *= nums[i]
return ans