@ 剑指offer（python）构建乘积数组

剑指offer刷题笔记51（python）

题目描述

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

思路

可以借助两个循环，第一个循环从上到下,依次计算B[i] = B[i-1]A[i-1]，第二个循环从下往上，引入一个temp元素，不断更新（依次累乘）temp = 1，1A[n-1]，1*A[n-1]*A[n-2]………………

代码

# -*- coding:utf-8 -*-
class Solution:  # 注意temp的使用
    def multiply(self, A):
        # write code here
        B = [0]*len(A)
        B[0] = 1
        for i in range(1,len(A)):
            B[i] = B[i-1]*A[i-1]
        temp = 1 
        for i in range(len(B)-2,-1,-1):
            temp *= A[i+1]
            B[i] *= temp
        return B