[LeetCode] 238. Product of Array Except Self @ python

tmfighting

于 2019-08-08 14:18:50 发布

阅读量236

点赞数

CC 4.0 BY-SA版权

分类专栏： leetcode 文章标签： leetcode

本文链接：https://blog.youkuaiyun.com/tmfighting/article/details/98868740

leetcode 专栏收录该内容

35 篇文章

订阅专栏

本文介绍了一种解决数组中每个元素等于其余元素乘积的问题的算法，不使用除法，时间复杂度为O(n)。通过预计算左右两边元素的乘积，巧妙地避免了直接除法运算，提供了两种实现方式，一种使用额外数组存储中间结果，另一种仅使用常数空间。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

一．题目：
给一个数组，求出每个位置的其他所有数的乘积．不能使用除法，时间复杂度要求是O(n).
Example:

Input:  [1,2,3,4]
Output: [24,12,8,6]

二．解题思路：
我们可以设计两个数组来储存每个位置的左边的数的积和右边的数的积．
代码如下：

class Solution(object):
    def productExceptSelf(self, nums):
        """
        :type nums: List[int]
        :rtype: List[int]
        """
        #时间复杂度 O(n)，空间复杂度O(n)
        a = [1]*len(nums)
        b = [1]*len(nums)
        for i in range(1,len(nums)):
            a[i] = a[i-1] * nums[i-1]
        for j in range(len(nums)-2,-1,-1):
            b[j] = b[j+1] * nums[j+1]
        
        return [a[i]*b[i] for i in range(len(nums))]

当然其实只需要常数空间也可以实现．代码如下：

class Solution(object):
    def productExceptSelf(self, nums):
        """
        :type nums: List[int]
        :rtype: List[int]
        """
        #时间复杂度 O(n)，空间复杂度O(1)
        result = [1]
        for i in range(1,len(nums)):
            result.append(result[-1]*nums[i-1])
        product =1
        for i in range(len(nums)-1,-1,-1):
            result[i] = product * result[i]
            product = nums[i]*product
        return result