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,解题思路
(1)计算B数组的下三角,
B
[
i
]
=
B
[
i
−
1
]
∗
A
[
i
−
1
]
B[i] = B[i-1]*A[i-1]
B[i]=B[i−1]∗A[i−1] 其中
i
∈
[
1
,
n
−
1
]
,
B
[
0
]
=
1
,
n
i\in[1,n-1],B[0] = 1, n
i∈[1,n−1],B[0]=1,n表示数组长度
(2)计算B数组的上三角,思路和(1)类似,初始化一个
t
e
m
p
=
1
temp = 1
temp=1,然后计算
t
e
m
p
∗
=
A
[
j
+
1
]
temp *= A[j+1]
temp∗=A[j+1]
B
[
j
]
∗
=
t
e
m
p
B[j] *= temp
B[j]∗=temp其中,
j
∈
[
n
−
2
,
0
]
j\in[n-2,0]
j∈[n−2,0]
3,源码
# -*- coding:utf-8 -*-
class Solution:
def multiply(self, A):
n = len(A)
B = [1]*n
for i in range(1,n):
B[i] = B[i-1] * A[i-1]
temp = 1
for j in range(n-2, -1, -1):
temp *= A[j+1]
B[j] *= temp
return B