题目描述
给定一个数组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]。不能使用除法。
思路
先算下三角,再算上三角。O(n)复杂度
class Solution:
def multiply(self, A):
# write code here
B = [1] * len(A)
temp = 1
for i in range(1, len(A)):
temp *= A[i-1]
B[i] *= temp
temp = 1
for i in range(len(A)-2, -1, -1):
temp *= A[i+1]
B[i] *= temp
return B
# [0, 1, 1]
# [1, 0, 1]
# [1, 1, 0]