数组——构建乘积数组(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,解题思路

(1)计算B数组的下三角,B[i]=B[i1]A[i1]B[i] = B[i-1]*A[i-1] 其中i[1,n1],B[0]=1,ni\in[1,n-1],B[0] = 1, n表示数组长度
(2)计算B数组的上三角,思路和(1)类似,初始化一个temp=1temp = 1,然后计算 temp=A[j+1]temp *= A[j+1] B[j]=tempB[j] *= temp其中,j[n2,0]j\in[n-2,0]

3,源码

# -*- coding:utf-8 -*-
class Solution:
    def multiply(self, A):
        n = len(A)
        B = [1]*n

        for i in range(1,n):
            B[i] = B[i-1] * A[i-1]

        temp = 1
        for j in range(n-2, -1, -1):
            temp *= A[j+1]
            B[j] *= temp
        return B
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章