数组——构建乘积数组（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]\ast A[i-1]$$ 其中$i\in [1,n-1],B[0]=1,n$表示数组长度
（2）计算B数组的上三角，思路和（1）类似，初始化一个$temp=1$，然后计算 $$temp\ast =A[j+1]$$ $$B[j]\ast =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

关注：挑灯看剑王知事