leetcode_238 除自身以外数组的乘积

1,058 阅读1分钟

要求

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

示例:

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

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

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

进阶: 你可以在常数空间复杂度内完成这个题目吗?( 出于对空间复杂度分析的目的,输出数组不被视为额外空间。)

核心代码

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

另一思路

class Solution:
    def productExceptSelf(self, nums: List[int]) -> List[int]:
        n = len(nums)
        res = [1] * n
        left = right = 1
        for l,r in zip(range(n),range(n-1,-1,-1)):
            res[l] *= left
            left *= nums[l]

            res[r] *= right
            right *= nums[r]
        return res

image.png

重点问题

解题思路:

第一种解法:我们使用左右累计的方式,以[1,2,3,4]为例,我们故意累乘到前一位,不会涉及到本位,一轮循环[1,1,2,6],我们在经历右侧累计,右侧第一个不能包括自己,使用1替换,[24,12,8,6],这种方法很巧妙,值得学习。第二种解法:同样是左右累计,注意zip的使用。