LeetCode#238. 除自身以外数组的乘积 (Python解题)

问题描述:

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

示例:

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

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

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

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/product-of-array-except-self

思路:

首先尝试了一种思路。对于每个元素对应的乘积结果,在一个循环里,用一个临时list集合储存除nums[i]之外的全部元素,使用reduce(lambda x,y:x*y,temp)来计算出对应的乘积,但是会超时。

然后调整思路:当前索引元素对应的乘积就是该元素左边所有元素的乘积乘以右边所有元素的乘积。

这种思路里面有几个要点:

①处理左边所有元素的乘积时应注意:lindex[0]应该是1,从左往右循环处理.需要在处理之前给初值,就需要初始化list。

②对于右边所有元素的乘积:rindex[n-1] 应该是1,从右往左循环处理。

③由于这道题目限制复杂度在O(n),因此使用单层循环来分别处理lindex[]、rindex[]、res[]就可以。

④注意循环的边界值

Python版本:

class Solution(object):
    def productExceptSelf(self, nums):
        """
        :type nums: List[int]
        :rtype: List[int]
        """
        n = len(nums)
        res = [0]*n
        lindex = [0]*n
        rindex = [0]*n
        lindex[0] = 1
        rindex[n-1] = 1
        for i in range(1,n):
            lindex[i] = lindex[i-1]*nums[i-1]
        for j in reversed(range(0,n-1)):
            rindex[j] = rindex[j+1]*nums[j+1]
        for k in range(0,n):
            res[k] = lindex[k]*rindex[k]
        return res

猜你喜欢

转载自blog.csdn.net/paranior/article/details/104403225
今日推荐