[238. 除自身以外数组的乘积]
难度 中等
给你一个长度为 n 的整数数组 nums
,其中 n > 1,返回输出数组 output
,其中 output[i]
等于 nums
中除 nums[i]
之外其余各元素的乘积。
示例:
输入: [1,2,3,4]
输出: [24,12,8,6]
**提示:**题目数据保证数组之中任意元素的全部前缀元素和后缀(甚至是整个数组)的乘积都在 32 位整数范围内。
说明: 请**不要使用除法,**且在 O(n) 时间复杂度内完成此题。
进阶:
你可以在常数空间复杂度内完成这个题目吗?( 出于对空间复杂度分析的目的,输出数组不被视为额外空间。)
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/product-of-array-except-self
解法一:左右乘积列表
乘积 = 当前数左边的乘积 * 当前数右边的乘积
我们用 left
和 right
分别存储数组中每个数字左边的乘积和右边的乘积, 其中
left[0] = 1
, left[i] = left[i - 1] * nums[i - 1]
right[nums.length - 1] = 1
, lright[i] = right[i + 1] * nums[i + 1]
则位置 i 处除自身以外数组的乘积 ans[i] = left[i] * right[i]
class Solution {
public int[] productExceptSelf(int[] nums) {
int len = nums.length;
// left 和 right 分别表示左右两侧的乘积列表
int[] left = new int[len];
int[] right = new int[len];
int[] ans = new int[len];
// 对于索引为 0 的元素,因为左侧没有元素,所以 left[0] = 1
left[0] = 1;
for (int i = 1; i < len; i++) {
left[i] = nums[i - 1] * left[i - 1];
}
// 对于索引为 len-1 的元素,因为右侧没有元素,所以 right[len-1] = 1
right[len - 1] = 1;
for (int i = len - 2; i >= 0; i--) {
right[i] = nums[i + 1] * right[i + 1];
}
// 对于索引 i,除 nums[i] 之外其余各元素的乘积 = 左侧所有元素的乘积 * 右侧所有元素的乘积
for (int i = 0; i < len; i++) {
ans[i] = left[i] * right[i];
}
return ans;
}
}
**改进:**使用上述 ans
数组来保存 left
和 right
class Solution {
public int[] productExceptSelf(int[] nums) {
int len = nums.length;
int[] ans = new int[len];
// 因为索引为 0 的元素左侧没有元素, 所以 ans[0] = 1
ans[0] = 1;
for (int i = 1; i < len; i++) {
ans[i] = nums[i - 1] * ans[i - 1];
}
// right 为右侧所有元素的乘积
// 刚开始右边没有元素,所以 right = 1
int right = 1;
for (int i = len - 1; i >= 0; i--) {
// 对于索引 i,左边的乘积为 ans[i],右边的乘积为 right
ans[i] = ans[i] * right;
//更新 right
right *= nums[i];
}
return ans;
}
}