16.除自身以外数组的乘积
大约 2 分钟
16.除自身以外数组的乘积
参考链接
238. 除自身以外数组的乘积 - 力扣(LeetCode)
238. 除自身以外数组的乘积 - 力扣(LeetCode)- 官方题解
个人尝试
class Solution {
public int[] productExceptSelf(int[] nums) {
// nums =[1, 2, 3, 4]
// prefix = [ 1, 1*1, 1*1*2, 1*1*2*3]
// suffix = [1*4*3*2, 1*4*3, 1*4, 1]
// output = prefix[i] * suffix[i]
int[] prefix = new int[nums.length];
// prefix = [ 1, 1*1, 1*1*2, 1*1*2*3]
prefix[0] = 1;
for (int i = 1; i < nums.length; i++) {
prefix[i] = prefix[i - 1] * nums[i - 1];
}
int[] suffix = new int[nums.length];
// suffix = [1*4*3*2, 1*4*3, 1*4, 1]
suffix[suffix.length - 1] = 1;
for (int i = suffix.length - 2; i >= 0; i--) {
suffix[i] = suffix[i + 1] * nums[i + 1];
}
for (int i = 0; i < nums.length; i++) {
nums[i] = prefix[i] * suffix[i];
}
return nums;
}
}
优秀题解
class Solution {
public int[] productExceptSelf(int[] nums) {
int length = nums.length;
// L 和 R 分别表示左右两侧的乘积列表
int[] L = new int[length];
int[] R = new int[length];
int[] answer = new int[length];
// L[i] 为索引 i 左侧所有元素的乘积
// 对于索引为 '0' 的元素,因为左侧没有元素,所以 L[0] = 1
L[0] = 1;
for (int i = 1; i < length; i++) {
L[i] = nums[i - 1] * L[i - 1];
}
// R[i] 为索引 i 右侧所有元素的乘积
// 对于索引为 'length-1' 的元素,因为右侧没有元素,所以 R[length-1] = 1
R[length - 1] = 1;
for (int i = length - 2; i >= 0; i--) {
R[i] = nums[i + 1] * R[i + 1];
}
// 对于索引 i,除 nums[i] 之外其余各元素的乘积就是左侧所有元素的乘积乘以右侧所有元素的乘积
for (int i = 0; i < length; i++) {
answer[i] = L[i] * R[i];
}
return answer;
}
}
作者:力扣官方题解
链接:https://leetcode.cn/problems/product-of-array-except-self/solutions/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。
优化空间复杂度:
class Solution {
public int[] productExceptSelf(int[] nums) {
int length = nums.length;
int[] answer = new int[length];
// answer[i] 表示索引 i 左侧所有元素的乘积
// 因为索引为 '0' 的元素左侧没有元素, 所以 answer[0] = 1
answer[0] = 1;
for (int i = 1; i < length; i++) {
answer[i] = nums[i - 1] * answer[i - 1];
}
// R 为右侧所有元素的乘积
// 刚开始右边没有元素,所以 R = 1
int R = 1;
for (int i = length - 1; i >= 0; i--) {
// 对于索引 i,左边的乘积为 answer[i],右边的乘积为 R
answer[i] = answer[i] * R;
// R 需要包含右边所有的乘积,所以计算下一个结果时需要将当前值乘到 R 上
R *= nums[i];
}
return answer;
}
}
作者:力扣官方题解
链接:https://leetcode.cn/problems/product-of-array-except-self/solutions/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。