数组——构建乘积数组（python实现）

1，问题描述

给定一个数组A[0,1,…,n-1],请构建一个数组B[0,1,…,n-1],其中B中的元素B[i]=A[0]A[1]…*A[i-1]A[i+1]…*A[n-1]。不能使用除法。

2，解题思路

（1）计算B数组的下三角，$B[i] = B[i-1]*A[i-1]$ 其中$i\in[1,n-1],B[0] = 1, n$表示数组长度
（2）计算B数组的上三角，思路和（1）类似，初始化一个$temp = 1$，然后计算 $temp *= A[j+1]$ $B[j] *= temp$其中，$j\in[n-2,0]$

3，源码

# -*- coding:utf-8 -*-
class Solution:
    def multiply(self, A):
        n = len(A)
        B = [1]*n

        for i in range(1,n):
            B[i] = B[i-1] * A[i-1]

        temp = 1
        for j in range(n-2, -1, -1):
            temp *= A[j+1]
            B[j] *= temp
        return B