链接:构建乘积数组
题目描述
给定一个数组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