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

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

中等题

题目

给你一个整数数组 nums，返回 数组 answer ，其中 answer[i] 等于 nums 中除 nums[i] 之外其余各元素的乘积 。

题目数据 保证 数组 nums之中任意元素的全部前缀元素和后缀的乘积都在 32 位 整数范围内。

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

示例1

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

示例2

输入: nums = [-1,1,0,-3,3]
输出: [0,0,9,0,0]

提示

• 2 <= nums.length <= 105
• -30 <= nums[i] <= 30
• 保证 数组 nums之中任意元素的全部前缀元素和后缀的乘积都在 32 位 整数范围内

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

解法

使用两个数组记录左右乘积列表，我们不必将所有数字的乘积除以给定索引处的数字得到相应的答案，而是利用索引左侧所有数字的乘积和右侧所有数字的乘积（即前缀与后缀）相乘得到答案。

class Solution:
    def productExceptSelf(self, nums: List[int]) -> List[int]:
        # answer[i]为左边乘积乘以右边乘积
        n = len(nums)
        left, right = [1 for _ in range(n)], [1 for _ in range(n)]
        for i in range(1,n): left[i] = left[i-1]*nums[i-1]
        for i in range(n-2,-1,-1): right[i] = right[i+1]*nums[i+1]
        for i in range(n):nums[i]=left[i]*right[i]
        return nums

• 时间复杂度 O(n)
• 空间复杂度 O(n)

相当于使用result来记录左边乘积，而后直接用常数记录右边乘积。

class Solution:
    def productExceptSelf(self, nums: List[int]) -> List[int]:
        # result[i]为左边乘积乘以右边乘积
        n = len(nums)
        result = [1 for _ in range(n)]
        for i in range(1,n): result[i] = result[i-1]*nums[i-1]
        tmp = 1
        for i in range(n-1,-1,-1): 
            result[i] *= tmp
            tmp *= nums[i]
        return result

• 时间复杂度 O(n)
• 空间复杂度 O(1)