238. 除自身以外数组的乘积
给你一个整数数组 nums,返回 数组 answer ,其中 answer[i] 等于 nums 中除 nums[i] 之外其余各元素的乘积 。
题目数据 保证 数组 nums之中任意元素的全部前缀元素和后缀的乘积都在 32 位 整数范围内。
请 不要使用除法,且在 O(n) 时间复杂度内完成此题。
示例 1:
输入: nums = [1,2,3,4]
输出: [24,12,8,6]
示例 2:
输入: nums = [-1,1,0,-3,3]
输出: [0,0,9,0,0]
提示:
2 <= nums.length <= 105
-30 <= nums[i] <= 30
保证 数组 nums之中任意元素的全部前缀元素和后缀的乘积都在 32 位 整数范围内
进阶:你可以在 O(1) 的额外空间复杂度内完成这个题目吗?( 出于对空间复杂度分析的目的,输出数组 不被视为 额外空间。)
解题
方法一:左右乘积列表
用两个数组记录 一个数 左侧、右侧的乘积,先初始化这两个数组,然后遍历得到答案。
// 时间O(n),空间O(n)
class Solution {
public int[] productExceptSelf(int[] nums) {
int n = nums.length;
// 记录对应位置数左侧的乘积,第一个数左侧乘积初始化为1
int[] l = new int[n];
l[0] = 1;
// 记录对应位置数右侧的乘积,最后一个数右侧乘积初始化为1
int[] r = new int[n];
r[n-1] = 1;
for (int i = 1; i < n; i++) {
l[i] = l[i-1] * nums[i-1];
}
for (int i = n-2; i >= 0; i--) {
r[i] = r[i+1] * nums[i+1];
}
int[] ans = new int[n];
for (int i = 0; i < n; i++) {
ans[i] = l[i]*r[i];
}
return ans;
}
}
方法一:左右乘积列表空间优化为 O(1)
题目中说:输出数组 不被视为 额外空间。
所以我们可以用输出数组代替左侧乘积数组l[],然后在遍历过程中用一个变量记录右侧乘积
// 时间O(n),空间O(1)
class Solution {
public int[] productExceptSelf(int[] nums) {
int n = nums.length;
// 记录对应位置数左侧的乘积,第一个数左侧乘积初始化为1
// 用输出数组代替左侧乘积数组l[]
int[] ans = new int[n];
ans[0] = 1;
for (int i = 1; i < n; i++) {
ans[i] = ans[i-1] * nums[i-1];
}
// 用一个变量记录右侧乘积,所以需要更新r,所以需要从右到左遍历
int r = 1;
for (int i = n-1; i >= 0; i--) {
ans[i] = ans[i]*r;
r = r*nums[i];
}
return ans;
}
}