3 构建乘积数组--剑指offer--python实现

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 解决方法

1. 逐个B[i]暴力求解
     对于每个元素B[i]求解一次，按照B[i]=A[0]A[1]…*A[i-1]A[i+1]…*A[n-1]。
   两重循环，时间复杂度为$O(n^2)$，空间复杂度为$O(1)$
2. 复用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(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