题目
- 给定一个数组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]。不能使用除法。
- leetcode链接:https://leetcode-cn.com/problems/gou-jian-cheng-ji-shu-zu-lcof/
思路
- 动态规划,维护左右两个乘积数组
B[i]的值可以看作下图的矩阵中每行的乘积。
下三角用连乘可以很容求得,上三角,从下向上也是连乘。
因此我们的思路就很清晰了,先算下三角中的连乘,即我们先算出B[i]中的一部分,然后倒过来按上三角中的分布规律,把另一部分也乘进去。
代码
/**
* @param {number[]} a
* @return {number[]}
*/
var constructArr = function (a) {
const leftDp = new Array(a.length).fill(1)
const rightDp = new Array(a.length).fill(1)
const result = []
for (let i = 1; i < a.length; i++) {
leftDp[i] = leftDp[i - 1] * a[i - 1]
}
for (let i = a.length - 2; i >= 0; i--) {
rightDp[i] = rightDp[i + 1] * a[i + 1]
}
for (let i = 0; i < a.length; i++) {
result.push(leftDp[i] * rightDp[i])
}
return result
}
/**
* @param {number[]} a
* @return {number[]}
* 不使用额外内存空间
*/
var constructArr = function(a) {
const result = new Array(a.length).fill(1)
let left = 1
let right = 1
for (let i = 1; i < a.length; i++) {
left *= a[i - 1]
result[i] *= left
}
for (let i = a.length - 2; i >= 0; i--) {
right *= a[i + 1]
result[i] *= right
}
return result
};