【主要考察点】:前缀和
解题方法
1. 暴力法
当做循环数组的方式,直接使用两层循环进行遍历,获得最终结果,但不满足时间复杂度O(n)的要求。
class Solution {
public int[] productExceptSelf(int[] nums) {
int len = nums.length;
int[] arr = new int[len];
int sum = 1;
for(int i = 0; i<len; i++){
sum = 1;
for(int j = 0; j<len-1; j++){
sum = sum * (nums[(j+i+1)%len]);
}
arr[i] = sum;
}
return arr;
}
}
2. 前缀和
我们可以分别计算每个下标位置,它的所有左侧元素的乘积、以及右侧元素的乘积,最后只要左右再相乘,得到的自然就是最终的结果。
那如何得到每个下标位置左右侧的元素乘积呢?就这可以通过前缀和的方式来解决了。
class Solution {
public int[] productExceptSelf(int[] nums) {
int len = nums.length;
int[] left = new int[len];
int[] right = new int[len];
left[0] = 1;
right[len-1] = 1;
// 从左向右遍历,计算每个下标位置的前缀乘积
for(int i = 1; i < len; i++){
left[i] = left[i - 1] * nums[i - 1];
}
// 从右向左遍历,计算每个下标位置的前缀乘积
for(int i = len - 2; i >= 0; i--){
right[i] = right[i + 1] * nums[i + 1];
}
int[] arr = new int[len];
for(int i = 0; i < len; i++){
arr[i] = left[i] * right[i];
}
return arr;
}
}
3. 进阶要求解法
本题的进阶要求不使用额外的空间来完成,我们还是用前缀乘积的方式,在不构建数组的情况下,也是可以完成的。
先构建一个左侧元素的前缀乘积,然后边计算右侧元素的前缀乘积,边计算当前下标结果。
class Solution {
public int[] productExceptSelf(int[] nums) {
int len = nums.length;
int[] arr = new int[len];
arr[0] = 1;
// 计算左侧元素的前缀乘积
for(int i = 1; i<len; i++){
arr[i] = arr[i - 1] * nums[i - 1];
}
// 用于维护右侧元素的前缀乘积
int right = 1;
for(int i = len - 1; i >= 0; i--){
// 直接计算左侧元素乘以右侧元素的结果
arr[i] = arr[i] * right;
right = right * nums[i];
}
return arr;
}
}