LeetCode238. 除自身以外数组的乘积（Python）

1、题目描述

https://leetcode-cn.com/problems/product-of-array-except-self/

给你一个长度为 n 的整数数组 nums，其中 n > 1，返回输出数组 output ，其中 output[i] 等于 nums 中除 nums[i] 之外其余各元素的乘积。

输入: [1,2,3,4]

输出: [24,12,8,6]

• 提示：题目数据保证数组之中任意元素的全部前缀元素和后缀（甚至是整个数组）的乘积都在 32 位整数范围内。
• 说明: 请不要使用除法，且在 O(n) 时间复杂度内完成此题。
• 进阶：你可以在常数空间复杂度内完成这个题目吗？（ 出于对空间复杂度分析的目的，输出数组不被视为额外空间。）

2、代码详解

技巧：走两遍for循环，用两个数组left和right记录左右数字的乘积

乘积 = 当前数左边的乘积 * 当前数右边的乘积

• 时间复杂度：O(N)，其中 N 指的是数组 nums 的大小。
• 空间复杂度：O(1)，输出数组不算进空间复杂度中，因此我们只需要常数的空间存放变量。

class Solution(object):
    def productExceptSelf(self, nums):
        """
        :type nums: List[int]
        :rtype: List[int]
        """
        n = len(nums)
        res = [0] * n
        left = 1
        for i in range(n):
            res[i] = left  # 此时数组存储的是除去当前元素左边的元素乘积
            left *= nums[i]
        right = 1
        for i in reversed(range(n)):  # n-1 ~ 0
            res[i] *= right  # 此时数组等于左边的 * 该数右边的
            right *= nums[i]  # k为该数右边的乘积

        return res