题目描述:
给定一个数组 A[0,1,…,n-1],请构建一个数组 B[0,1,…,n-1],其中 B[i] 的值是数组 A 中除了下标 i 以外的元素的积, 即 B[i]=A[0]×A[1]×…×A[i-1]×A[i+1]×…×A[n-1]。不能使用除法。
示例:
输入: [1,2,3,4,5]
输出: [120,60,40,30,24]
思路1:
按照题目中的要求直接计算,但是这样会进行大量重复的计算,时间复杂度会很高。
思路2:
充分考虑会重复计算的部分,可以得到这样的图示:
所以分成上三角和下三角两个部分进行计算,这样遍历两遍就可以完成,时间复杂度为O(n)。
具体思路参考的图解链接
class Solution:
def constructArr(self, a: List[int]) -> List[int]:
b, temp = [1] * len(a), 1
for i in range(1, len(b)):
b[i] = b[i-1] * a[i-1]
for j in range(len(b)-2, -1, -1):
temp *= a[j+1]
b[j] *= temp
return b
其实分成的两个三角形求解部分相当于有两个dp数组在求。