(python版)《剑指Offer》JZ51：构建乘积数组

牛客
Leetcode

【思路】对称遍历

根据表格的主对角线（全为 1 ），可将表格分为 上三角 和 下三角 两部分。分别迭代计算下三角和上三角两部分的乘积

算法流程：

1. 初始化：数组 B ，其中 B[0] = 1 ；辅助变量 tmp = 1 ；
2. 计算 B[i] 的 下三角 各元素的乘积，直接乘入 B[i]；
3. 计算 B[i] 的 上三角 各元素的乘积，记为 tmp ，并乘入 B[i] ；
4. 返回 B 。
   

a=[1,2,3,4,5]
b[0]=1	tmp=1
i=0
b[1]=b[0]*a[0]=1*1=1
b[2]=b[1]*a[1]=1*2=2
b[3]=b[2]*a[2]=2*3=6
b[4]=b[3]*a[3]=6*4=24
b[5]=b[4]*a[4]=24*5=120

tmp=tmp*a[3+1]=1*5=5		b[3]=b[3]*tmp=6x5=30
tmp=tmp*a[2+1]=5*4=20		b[2]=b[2]*tmp=2x20=40
tmp=tmp*a[1+1]=20*3=60		b[1]=b[1]*tmp=1x60=60
tmp=tmp*a[0+1]=60*2=120		b[0]=b[0]*tmp=120x1=120

class Solution:
    def constructArr(self, a: List[int]) -> List[int]:
        b, tmp = [1] * len(a), 1
        for i in range(1, len(a)):
            b[i] = b[i - 1] * a[i - 1] # 下三角
        for i in range(len(a) - 2, -1, -1): 
            tmp *= a[i + 1] # 上三角
            b[i] *= tmp # 下三角 * 上三角
        return b
'''
作者：jyd
链接：https://leetcode-cn.com/problems/gou-jian-cheng-ji-shu-zu-lcof/solution/mian-shi-ti-66-gou-jian-cheng-ji-shu-zu-biao-ge-fe/
来源：力扣（LeetCode）
'''

• 时间复杂度 O(N) ： 其中 N 为数组长度，两轮遍历数组 a ，使用 O(N) 时间。
• 空间复杂度 O(1) ： 变量 tmp 使用常数大小额外空间（数组 b 作为返回值，不计入复杂度考虑）。