这道题目要求计算一个数组中除了当前位置之外的所有元素的乘积数组。给定一个整数数组 nums
,我们需要返回一个新的数组,其中每个元素是原数组中除了自身之外所有元素的乘积。
解决这个问题的一种常见方法是使用前缀和和后缀和的思想。我们可以先计算每个位置左侧所有元素的乘积,然后再计算每个位置右侧所有元素的乘积。最后将左侧乘积和右侧乘积相乘,就得到了所需的结果。
下面是对给定代码的详细解释:
class Solution {
public int[] productExceptSelf(int[] nums) {
int length = nums.length;
int[] res = new int[length]; // 创建结果数组
res[0] = 1; // 初始值为1,因为第一个元素左侧没有元素
// 计算每个位置左侧所有元素的乘积
for (int i = 1; i < res.length; i++) {
res[i] = res[i - 1] * nums[i - 1];
}
int right = 1; // 初始值为1,因为最后一个元素右侧没有元素
// 计算每个位置右侧所有元素的乘积,并与左侧乘积相乘
for (int i = length - 1; i >= 0; i--) {
res[i] = res[i] * right;
right = right * nums[i];
}
return res; // 返回结果数组
}
}
这个解法的时间复杂度为 O(n),其中 n 是数组 nums
的长度。我们需要遍历两次数组来计算左侧乘积和右侧乘积。
首先,我们初始化结果数组 res
,并将第一个元素设置为1,表示左侧乘积。然后,我们从左到右遍历数组,计算每个位置左侧所有元素的乘积,并将结果保存在 res
数组中。
接下来,我们使用一个变量 right
来表示右侧乘积。从右到左遍历数组,计算每个位置右侧所有元素的乘积,并将其与左侧乘积相乘,更新 res
数组。
最后,返回结果数组 res
。