LeetCode:238. 除自身以外数组的乘积(python)

LeetCode:238. 除自身以外数组的乘积(python)

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

示例:

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

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

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

LeetCode 链接

分析

  • 根据题目要求,需要计算当前位置的左侧累积×右侧累积
  • 时间复杂度要求 O ( n ) O(n) O(n),可分别记录左右侧累积,方便计算

思路

  • 通过两次遍历,第一次从左往右遍历更新左侧累积;第二次从右往左遍历,乘右侧累积并更新右侧累积(代码1
  • 使用左右指针,通过一次遍历,分别乘左右累积并更新左右累积
附代码1(Python):
class Solution:
    def productExceptSelf(self, nums):
        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
test = Solution()
nums = [1,2,3,4]
test.productExceptSelf(nums)
[24, 12, 8, 6]
附代码2(Python):
class Solution:
    def productExceptSelf(self, nums):
        n = len(nums)
        res = [1] * n         # 初始化
        left = right = 1      # 记录左侧累积和右侧累积
        for l, r in zip(range(n), range(n-1, -1, -1)):   # l,左指针;r,右指针
            res[l] *= left    # ×左侧累积
            left *= nums[l]   # 更新左侧累积
            
            res[r] *= right   # ×右侧累积
            right *= nums[r]  # 更新右侧累积
        return res
test = Solution()
nums = [1,2,3,4]
test.productExceptSelf(nums)
[24, 12, 8, 6]

你可能感兴趣的:(LeetCode,LeetCode,238.,除自身以外数组的乘积,python)