leetcode238.除自身以外数组的乘积 Python

题目:

给你一个长度为 n 的整数数组 nums,其中 n > 1,返回输出数组 output ,其中 output[i] 等于 nums 中除 nums[i] 之外其余各元素的乘积。
说明: 请不要使用除法,且在 O(n) 时间复杂度内完成此题。

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

示例:

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

思路:

建立一个结果数组,初始化为1。

然后先从前往后遍历,第i项代表前i个元素的乘积,不包括i位置的元素,然后再反向遍历一遍,从后往前遍历,在当前结果数组的基础上,乘以当前位置前面的数,不包括当前位置的数,然后返回这个结果数组,就是不包括自身以外的数组的乘积。

class Solution(object):
    def productExceptSelf(self, nums):
        """
        :type nums: List[int]
        :rtype: List[int]
        """
        dp = [1] * len(nums)
        for i in range(1,len(nums)):
            dp[i] = dp[i-1] * nums[i-1]
        prod = nums[len(nums)-1]
        for j in range(len(nums)-2,-1,-1):
            dp[j] = dp[j] * prod
            prod *= nums[j]
        return dp 

版权声明:本文为qq_36849855原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。