剑指offer刷题笔记-66. 构建乘积数组

在这里插入图片描述
又是一道思路非常简单,但是操作时需要技巧的题。基本思路就是每个元素B[i] = A[1:i-1]的阶乘*A[i+1:]的阶乘。为了避免暴力法求B中每个元素时都要算两遍阶乘,可以事先把A从左到右的阶乘和从右到左的阶乘算出来并储存。
代码如下:

class Solution:
    def constructArr(self, a: List[int]) -> List[int]:
        left = []
        right = []
        prod_l = 1
        prod_r = 1
        b = []
        for i in range(len(a)):
            prod_l *= a[i]
            left.append(prod_l)
            prod_r *= a[len(a)-i-1]
            right.append(prod_r)
        if len(a)<2: return []
        b.append(right[-2])
        for j in range(1,len(a)-1):
            b.append(left[j-1]*right[len(a)-j-2])
        b.append(left[-2])
        return b

我的这段代码非常不简练,其实right和left两个数组都可以简化去掉, 代码如下:

class Solution:
    def constructArr(self, a: List[int]) -> List[int]:
        b, tmp = [1] * len(a), 1
        for i in range(1, len(a)):  #1 to n-1
            b[i] = b[i - 1] * a[i - 1] # 左边的阶乘直接储存在b中
        for i in range(len(a) - 2, -1, -1):  #n-2 to 0
            tmp *= a[i + 1] # 右边阶乘
            b[i] *= tmp

你以为到这就是巅峰了?那你就太小看leetcode上的大神了,整个过程其实可以合并到一个循环中,比如在同一个循环中可以计算b[1]乘以右边阶乘和b[n-2]乘以左边阶乘,不必拘泥于一个循环中只能指针i所指向的位置:

class Solution:
    def constructArr(self, a: List[int]) -> List[int]:
        b = [1] * len(a)
        left, right = 1, 1
        for i in range(1,len(a)):
            left *= a[i-1]
            b[i] *= left
            right *= a[len(a)-i]
            b[len(a)-i-1] *= right
        return b

不过这种方法虽然简练但是速度不如第二种方法,看来第二种还是巅峰XD

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值