51.构建乘积数组

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.思路

如果没有不能使用除法的限制，可以直接用累乘的结果除以A[i]。由于题目有限制，一种直观的解法是连乘n-1个数字，但时间复杂度是O(n^2)。可以把B[i]=A[0]A[1]…*A[i-1]A[i+1]…*A[n-1]分成A[0]A[1]…A[i-1]和A[i+1]…*A[n-1]两部分的乘积。

3.代码

class Solution:
    def multiply(self, A):
        C,D = [],[]
        for i in range(len(A)):
            if i == 0:
                C.append(1)
            else:
                C.append(C[i-1]*A[i-1])
        for i in range(len(A)):
            if i == 0:
                D.append(1)
            else:
                D.append(D[i-1]*A[len(A)-i])
        D = D[::-1]
        B = []
        for i in range(len(A)):
            B.append(C[i]*D[i])
        return B