LeetCode238. 除自身以外数组的乘积(python)

给定长度为 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

 

你可能感兴趣的:(LeetCode)