數組——構建乘積數組(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
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章