【剑指offer python】面试题52：构建乘积数组

链接：构建乘积数组

题目描述

给定一个数组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]。不能使用除法。

解法一：去除A中序号为i的元素，其他元素乘积就是B[i]

class Solution:
    def multiply(self, A):
        B = [1]*len(A)
        for i in range(0,len(A)):
            for j in range(0,len(B)):
                if i!=j:
                    B[i] *= A[j]
        return B

解法二：先求A[0]*...A[i-1]，再求A[i+1]*...A[n]，最后乘在一起。

class Solution:
    def multiply(self, A):
        B = [1]*len(A)
        temp = 1
        for i in range(len(A)):
            temp *= A[i-1] if i > 0 else temp
            for j in A[i+1:]:
                B[i] *= j
            B[i] = temp * B[i]
        return B