题目:除自身以外数组的乘积
给你一个整数数组 nums
,返回 数组 answer
,其中 answer[i]
等于 nums
中除 nums[i]
之外其余各元素的乘积 。
题目数据 保证 数组 nums
之中任意元素的全部前缀元素和后缀的乘积都在 32 位 整数范围内。
请 不要使用除法,且在 O(n)
时间复杂度内完成此题。(如果可以使用除法,就是所有元素之积除以当前元素)
思路:最后的结果为当前元素的前缀之积《乘以》后缀之积。其中,计算前缀之积第1个元素的前缀之积为1,计算后缀之积,最后一个元素的后缀之积为1,当前元素的前缀之积等于前一个元素乘以前一个元素的前缀之积,当前元素的后缀之积等于后一个元素乘以后一个元素的后缀之积。
代码:
from typing import List
class Solution:
def productExceptSelf(self, nums: List[int]) -> List[int]:
len_nums = len(nums)
result = [0] * len_nums
result[0] = 1
for i in range(1, len_nums):
result[i] = result[i - 1] * nums[i - 1] # 前缀之积
# print(result)
tmp = 1
for j in range(len_nums - 2, -1, -1):
tmp *= nums[j + 1]
# print("tmp", tmp)
result[j] = tmp * result[j]
# print("result[j]", result[j])
# print(result)
return result
num = [1, 2, 3, 4]
s1 = Solution()
s1.productExceptSelf(num)