剑指offer第二版（Python3）--面试题66 :构建乘积数组

第2章 面试需要的基础知识

第3章 高质量的代码

第4章 解决面试题的思路

第5章 优化时间和空间效率

第6章 面试中的各项能力

  面试题62 : 圆圈中最后剩下的数字

  面试题64 : 求1+2+3+…+n

  面试题65 : 不用加减乘除做加法

  面试题66 :构建乘积数组

第7章 两个面试案例

---

题目描述
牛客网
  给定一个数组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)。
  可以把B[i]=A[0]A[1]…A[i-1]A[i+1]…A[n-1]看成A[0]A[1]…A[i-1]和A[i+1]…A[n-2]A[n-1]两部分的乘积。即通过A[i]项将B[i]分为两部分的乘积。效果相当于是个对角矩阵。
        C[i] = A[0]A[1]…A[i-1]
        D[I] = A[i+1]…A[n-1]
  所以有B[i] = C[i] * D[i]

  B[i]的左半部分(红色部分)和B[i-1]有关（将B[i]的左半部分乘积看成C[i]，有C[i]=C[i-1]*A[i-1]），

  B[i]的右半部分(紫色部分)与B[i+1]有关（将B[i]的右半部分乘积看成D[i]，有D[i]=D[i+1]*A[i+1]）

实战

class Solution:
    def multiply(self, A):
        # write code here
        if not A:
            return None
        
        n = len(A)
        C = [None] * n
        D = [None] * n
        C[0] = 1
        D[n-1] = 1
        for i in range(1, n):
            C[i] = C[i-1] * A[i-1]
        for i in range(n-2, -1, -1):
            D[i] = D[i+1] * A[i+1]
        
        for i in range(n):
            C[i] = C[i] * D[i]
        return C

简化版

class Solution:
    def multiply(self, A):
        # write code here
        if not A:
            return None
        
        n = len(A)
        C = [None] * n
        C[0] = 1
        temp = 1
        for i in range(1, n):
            C[i] = C[i-1] * A[i-1]
        for i in range(n-2, -1, -1):
            temp *= A[i+1]
            C[i] *= temp
        return C