剑指offer面试题66. 构建乘积数组

题目描述

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

思路

详见链接

代码

class Solution:
	def constructArr(self, a; List[int])->List[int]:
		b, tmp = [1]*len(a), 1
		for i in range(1, len(a)):
			b[i] = b[i-1]*a[i-1]
		for i in range(len(a)-2, -1, -1):
			tmp *= a[i+1]
			b[i] *= tmp
		return b

复杂度

时间复杂度 O(N)： 其中 NN 为数组长度，两轮遍历数组 a ，使用 O(N)时间。
空间复杂度 O(1) ： 变量 tmp 使用常数大小额外空间（数组 b 作为返回值，不计入复杂度考虑）。