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. 题目解析
2.1 考察问题
本问题考察的是数组遍历技巧的问题。
可以看到数组B元素的规律,涉及到求解A数组内部元素的连乘,连乘包括前半部分和后半部分,最容易想到的就是B[i]求解一次。但是时间复杂度过高,优化方法是能够复用A的连乘,而不是每次重新再乘。
对于复用问题,很可能想到A的所有元素乘积,然后对应除以一下A[i]得到B[i],但是必须要注意到A中的元素并没有说明不为零,因此不能用除法。
2.2 解决方法
- 逐个B[i]暴力求解
对于每个元素B[i]求解一次,按照B[i]=A[0]A[1]…*A[i-1]A[i+1]…*A[n-1]。
两重循环,时间复杂度为 O ( n 2 ) O(n^2) O(n2),空间复杂度为 O ( 1 ) O(1) O(1) - 复用A的连乘求解
考虑先从左到右遍历一次A,将B[i]乘积的左半部分A[0]A[1]…A[i-1]求出来。然后再从右到左遍历A求A[i+1]…*A[n-1],然后与对应的B[i]相乘,就得到了最终结果。
2.3 优化分析
暴力求解给出了时间复杂度为
O
(
n
2
)
O(n^2)
O(n2),空间复杂度为
O
(
1
)
O(1)
O(1)的解法,显然优化要从时间复杂度方面入手。
是什么导致了这么高的时间复杂度,是两重遍历。那么是否可以减少两重遍历,是否有冗余的工作量可以优化?想到了复用A的连乘。
我想在算法解决问题过程中,以上优化思路是常见的,要有理有据的进行,而不是胡思乱想碰运气。
3. 代码实现
3.1 暴力求解
class Solution:
def multiply(self, A):
# write code here
# 不能使用除法,使用除法可能会遇到除0问题
B = [1] * len(A)
for i in range(0, len(A)):
for j in range(0, len(A)):
if i != j:
B[i] = B[i] * A[j] ## 每个B[i]求解一次
return B
3.2 复用A的连乘求解
class Solution:
def multiply(self, A):
# write code here
# 不能使用除法,使用除法可能会遇到除0问题
B = [1] * len(A)
for i in range(1, len(A)):### 求解A[0]-A[i-1]的连乘
B[i] = B[i - 1]*A[i-1]
temp = 1
for i in range(len(A)-2, -1, -1):### 倒序遍历A,求解A[i+1]-A[n-1]的连乘
temp = temp * A[i+1]
B[i] = B[i] * temp
return B