238. 除自身以外数组的乘积
20200604
难度:中等
题目描述
给你一个长度为 n 的整数数组 nums,其中 n > 1,返回输出数组 output ,其中 output[i] 等于 nums 中除 nums[i] 之外其余各元素的乘积。
示例:
输入: [1,2,3,4]
输出: [24,12,8,6]
提示:
题目数据保证数组之中任意元素的全部前缀元素和后缀(甚至是整个数组)的乘积都在 32 位整数范围内。
说明:
请不要使用除法,且在 O(n) 时间复杂度内完成此题。
进阶:
你可以在常数空间复杂度内完成这个题目吗?( 出于对空间复杂度分析的目的,输出数组不被视为额外空间。)
Solution
方法一
索引左侧所有数字的乘积和右侧所有数字的乘积(即前缀与后缀)相乘
class Solution {
public int[] productExceptSelf(int[] nums) {
int length = nums.length;
int[] L = new int[length];
int[] R = new int[length];
int[] ans = new int[length];
// L[i] 为索引 i 左侧所有元素的乘积
L[0] = 1;
for(int i = 1; i < length; i++){
L[i] = nums[i-1] * L[i-1];
}
// R[i] 为索引 i 右侧所有元素的乘积
R[length-1] = 1;
for(int i = length-2; i >= 0; i--){
R[i] = nums[i+1] * R[i+1];
}
// 左侧所有元素的乘积乘以右侧所有元素的乘积
for(int i = 0; i < length; i++){
ans[i] = L[i] * R[i];
}
return ans;
}
}
复杂度分析
- 时间复杂度:O(N)
- 空间复杂度:O(N)
方法二
不构造 R
数组。而是用一个遍历来跟踪右边元素的乘积
class Solution {
public int[] productExceptSelf(int[] nums) {
int length = nums.length;
int[] answer = new int[length];
// answer[i] 表示索引 i 左侧所有元素的乘积
answer[0] = 1;
for (int i = 1; i < length; i++) {
answer[i] = nums[i - 1] * answer[i - 1];
}
// R 为右侧所有元素的乘积
int R = 1;
for (int i = length - 1; i >= 0; i--) {
// 左边的乘积为 answer[i],右边的乘积为 R
answer[i] = answer[i] * R;
R *= nums[i];
}
return answer;
}
}