题目:
给你一个整数数组 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]
提示:
2 <= nums.length <= 105
-30 <= nums[i] <= 30
保证 数组 nums之中任意元素的全部前缀元素和后缀的乘积都在 32 位 整数范围内
进阶:你可以在 O(1) 的额外空间复杂度内完成这个题目吗?( 出于对空间复杂度分析的目的,输出数组不被视为额外空间。)
代码:
代码如下(示例):
from typing import List
class Solution:
def productExceptSelf(self, nums: List[int]) -> List[int]:
n = len(nums)
res = [1] * n
for i in range(1, n):#先从前面乘到后面
res[i] = res[i - 1] * nums[i - 1]
right = 1
for i in range(n - 1, -1, -1):#再从后面乘到前面
res[i] *= right
right *= nums[i]
return res
if __name__ == '__main__':
s = Solution()
print(s.productExceptSelf([1, 2, 3, 4]))
print(s.productExceptSelf([-1, 1, 0, -3, 3]))
print(s.productExceptSelf([0, 0]))
print(s.productExceptSelf([1, 2, 3, 4, 5]))
print(s.productExceptSelf([1, 2, 3, 4, 5, 6]))
print(s.productExceptSelf([1, 2, 3, 4, 5, 6, 7]))
print(s.productExceptSelf([1, 2, 3, 4, 5, 6, 7, 8]))
print(s.productExceptSelf([1, 2, 3, 4, 5, 6, 7, 8, 9]))
print(s.productExceptSelf([1, 2, 3, 4, 5, 6, 7, 8, 9, 10]))
print(s.productExceptSelf([1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11]))