题目描述
给定一个数组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]。不能使用除法。
解题思路
《剑指offer》的思路:
B[i]的值可以看作下图的矩阵中每行的乘积。下三角用连乘可以很容求得,上三角,从下向上也是连乘。
因此:先算下三角中的连乘,即我们先算出B[i]中的一部分,然后倒过来按上三角中的分布规律,把另一部分也乘进去。
def multiply(self, A):
mult = 1
B = [1]*len(A)
for i in range(len(A)):
mult = mult * A[i-1] if i>0 else mult
for a in A[i+1:]:
B[i] = B[i] * a
B[i] = B[i] * mult
return B
改进一点:
def multiply(self, A):
mult = 1
B = [1]*len(A)
for i in range(len(A)):
mult = mult * A[i-1] if i>0 else mult
for a in A[i+1:]:
B[i] = B[i] * a
B[i] = B[i] * mult
return B