Leetcode 238.除自身以外数组的乘积(medium)

题目

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

示例:

输入: [1,2,3,4]
输出: [24,12,8,6]
 

提示：题目数据保证数组之中任意元素的全部前缀元素和后缀（甚至是整个数组）的乘积都在 32 位整数范围内。

说明: 请不要使用除法，且在 O(n) 时间复杂度内完成此题。

进阶：
你可以在常数空间复杂度内完成这个题目吗？（ 出于对空间复杂度分析的目的，输出数组不被视为额外空间。）

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/product-of-array-except-self
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

解题思路

1. 原始解法

 题目的难点在于不允许使用除法，且要在O(n)内完成。通过学习题解，得到思路：维护两个数组，记录每一个元素左右两侧的乘积和即可。

结果：通过测试用例，用时76ms，内存使用量20.9MB

代码：

class Solution:
    def productExceptSelf(self, nums: List[int]) -> List[int]:
        length = len(nums)
        left = [1 for i in range(length)]
        right = [1 for i in range(length)]

        for i in range(1, length):
            left[i] = left[i - 1] * nums[i - 1]

        for i in range(length - 2, -1, -1):
            right[i] = right[i + 1] * nums[i + 1]

        output = [0 for i in range(length)]
        for i in range(length):
            output[i] = left[i] * right[i]

        return output

2. 第一次优化

我们的原始解法里有三次循环，考虑到最后输出的时候使用的循环，我们可以不必记录left数组，转而记录一个left参数即可。计算完output[i]之后，将left乘以nums[i]，即是下一个数的左侧元素的乘积。

结果：通过测试用例，用时68ms，内存使用量20.4MB

class Solution:
    def productExceptSelf(self, nums: List[int]) -> List[int]:
        length = len(nums)
        left = 1
        right = [1 for i in range(length)]
        output = [0 for i in range(length)]

        for i in range(length-2, -1, -1):
            right[i] = right[i + 1] * nums[i + 1]            

        
        for i in range(length):            
            output[i] = left * right[i]
            left = left * nums[i]

        return output

3. 第二次优化

其实我们可以将right数组内的元素直接存储在output中，这个可以进一步减少空间复杂度。

结果：通过测试用例，用时60ms，内存使用量19.6MB

代码：

class Solution:
    def productExceptSelf(self, nums: List[int]) -> List[int]:
        length = len(nums)
        left = 1
        output = [1 for i in range(length)]

        for i in range(length-2, -1, -1):
            output[i] = output[i + 1] * nums[i + 1]            

        
        for i in range(length):            
            output[i] = left * output[i]
            left = left * nums[i]

        return output