给定长度为 n 的整数数组 nums,其中 n > 1,返回输出数组 output ,其中 output[i] 等于 nums 中除 nums[i] 之外其余各元素的乘积。
示例:
输入: [1,2,3,4](nums)
输出: [24,12,8,6](output)
说明: 请不要使用除法,且在 O(n) 时间复杂度内完成此题
tips 不能使用循环嵌套(时间复杂度)
运算思路:
遍历两遍列表,第一次,记录位置i之前的乘积;第二次,记录i之后的乘积
first step:
output[0] = 1
output[1] = 1*1
output[2] = 1*2
output[3] = 1*2*3
second step:(倒序遍历nums)
output[3] = 1*2*3*1 = 6
output[2] = 1*2*(1*4) = 8
output[1] = 1*1*(1*4*3) = 12
output[0] = 1*(1*4*3*2) = 24
代码:
class Solution(object):
def productExceptSelf(self, nums):
"""
:type nums: List[int]
:rtype: List[int]
"""
output = []
out = 1
for i in range(len(nums)):
if i == 0:
output.append(1)
else:
out = out * nums[i-1]
output.append(out)
middle = 1
for i in range(len(nums)-1, -1, -1):
output[i] = output[i] * middle
middle = middle * nums[i]
return output