剑指Offer 66 构建乘积数组

构建乘积数组

给定一个数组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]。不能使用除法。

# -*- coding:utf-8 -*-
class Solution:
    def multiply(self, A):
        n = len(A)
        mat = [[1 for _ in range(n)]for _ in range(n)]
        for i in range(n):
            for j in range(n):
                if i != j:
                    mat[i][j] = A[j]
        B = [1] * n
        for i in range(n):
            mul = 1
            for j in range(n):
                mul *= mat[i][j]
            B[i] = mul
        return B
        # write code here

思路：先构建mat二维数组，这个矩阵的特点是，主对角线都是1。从每一行来看，除了在主对角线上的元素为1，其余元素与A对应位置相同。

利用上面的步骤建立好的二维数组，进行本题的计算，只需要将每一行所有元素相乘就可以得到B中的对应值。