LeetCode238 除自身以外数组的乘积 & 剑指Offer 66 构建乘积数组
题目
LeetCode238 除自身以外数组的乘积:
剑指 Offer 66. 构建乘积数组:
剑指 Offer 66 的区别在于输入数组可能为空。
解题
解题一:空间复杂度为O(n)
// javascript
var productExceptSelf = function(nums) {
if (nums.length === 0) return [];
const n = nums.length;
// 左右两侧的乘积列表
const multiL = new Array(n), multiR = new Array(n);
const ans = new Array(n);
// multiL[i] 为索引 i 左侧所有元素的乘积
// 对于索引为 '0' 的元素,因为左侧没有元素,所以 multiL[0] = 1
multiL[0] = 1;
for (let i = 1; i < n; i++) multiL[i] = nums[i - 1] * multiL[i - 1];
// multiR[i] 为索引 i 右侧所有元素的乘积
// 对于索引为 'n-1' 的元素,因为右侧没有元素,所以 multiR[n - 1] = 1
multiR[n - 1] = 1;
for (let i = n - 2; i >= 0; i--) multiR[i] = nums[i + 1] * multiR[i + 1];
// 对于索引 i,除 nums[i] 之外其余各元素的乘积就是左侧所有元素的乘积乘以右侧所有元素的乘积
for (let i = 0; i < n; i++) ans[i] = multiL[i] * multiR[i];
return ans;
};
解题二:空间复杂度为O(1)
// javascript
var productExceptSelf = function(nums) {
if (nums.length === 0) return [];
const n = nums.length;
// ans[i] 表示索引 i 左侧所有元素的乘积
const ans = new Array(n);
// 因为索引为 '0' 的元素左侧没有元素, 所以 ans[0] = 1
ans[0] = 1;
for (let i = 1; i < n; i++) ans[i] = nums[i - 1] * ans[i - 1];
// multiR 为右侧所有元素的乘积
// 刚开始右边没有元素,所以 multiR = 1
let multiR = 1;
for (let i = n - 1; i >= 0; i--) {
// 对于索引 i,左边的乘积为 ans[i],右边的乘积为 multiR
ans[i] *= multiR;
// multiR 需要包含右边所有的乘积,所以计算下一个结果时需要将当前值乘到 multiR 上
multiR *= nums[i];
}
return ans;
};