LeetCode 238 除自身以外数组的乘积

题目:

给你一个整数数组 nums,返回 数组 answer ,其中 answer[i] 等于 nums 中除 nums[i] 之外其余各元素的乘积 。

题目数据 保证 数组 nums之中任意元素的全部前缀元素和后缀的乘积都在 32 位 整数范围内。

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

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

思路:

简单方法:遍历一遍计算每个数的前缀积和后缀积,使用两个列表prefix和postfix储存每个数的前缀积和后缀积,最后遍历这两个列表计算结果res[i] = prefix[i] * postfix[i + 1]

图中同一颜色代表关系项
class Solution:
    def productExceptSelf(self, nums: List[int]) -> List[int]:
        length = len(nums)
        x = y = 1
        prefix = [1]
        postfix = [1]
        for i in range(length):
            x *= nums[i]
            y *= nums[length - i - 1]
            prefix.append(x)
            postfix.append(y)
        postfix.reverse()

        res = []
        for i in range(length):
            res.append(prefix[i] * postfix[i + 1])
        return res

进阶方法:根据题目题目进阶要求,要使用O(1)的额外空间,就不能使用两个列表来储存计算出的前后缀积,这时候就要转变一下思路。

已知前缀级必须由前往后遍历,后缀积必须由后向前遍历,且res[i] = prefix[i] * postfix[i + 1]

 令res[i]初始为1,按照循环顺序从res的角度看,

res[i] *= 当前循环的prefix,

res[length - i - 1] *= 当前循环的postfix;

从prefix和postfix角度看,prefix*= nums[i],postfix *= nums[length - i - 1]

则直接遍历一次使用两个变量prefix和postfix即可得出答案res

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

        

 此时空间效率已为O(1)

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值