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[i−1]∗A[i−1]B[i] = B[i-1]*A[i-1]B[i]=B[i−1]∗A[i−1] 其中i∈[1,n−1],B[0]=1,ni\in[1,n-1],B[0] = 1, ni∈[1,n−1],B[0]=1,n表示数组长度
(2)计算B数组的上三角,思路和(1)类似,初始化一个temp=1temp = 1temp=1,然后计算 temp∗=A[j+1]temp *= A[j+1]temp∗=A[j+1] B[j]∗=tempB[j] *= tempB[j]∗=temp其中,j∈[n−2,0]j\in[n-2,0]j∈[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