题解
- 题意:给你一个长度为 n 的整数数组 nums,其中 n > 1,返回输出数组 res,其中 res[i] 等于 nums 中除 nums[i] 之外其余各元素的乘积。
输入: [1,2,3,4]
输出: [24,12,8,6]
- 题解:想要题目的意思,我们可以考虑对每个数字求它左边和右边的数据的乘积。因此最直观的做法就是利用一个
L
数组保留第i
个数字之前的所有数字的乘积,利用R
数组保留第i
个数字之后所有数字的乘积,然后再遍历一次,让res[i]=L[i]*R[i]
即可 - 实现
class Solution {
public int[] productExceptSelf(int[] nums) {
int n = nums.length;
int []L = new int[n];
int []R = new int[n];
L[0] = 1;
R[n-1] = 1;
for(int i = 1;i < n;i ++){
L[i] = L[i-1]*nums[i-1];
R[n-1-i] = R[n-i]*nums[n-i];
}
int []res = new int[n];
for(int i = 0;i < n;i++){
res[i] = L[i]*R[i];
}
return res;
}
}
- 但是其实并不需要两个数组来存储信息,我们可以直接使用
res
数组正向遍历一遍,利用一个变量累计从左向右的乘积,然后再反向遍历一遍,再利用另一个变量累计从右向左的乘积,这个过程可以看成利用两个直角三角形构造一个长为n,宽为(n-1)
的矩形的过程
class Solution {
public int[] productExceptSelf(int[] nums) {
int n = nums.length;
int left = 1;
int right = 1;
int []res = new int[n];
res[0] = 1;
for(int i = 1;i < n;i ++){
left = nums[i-1]*left;
res[i] = left;
}
for(int i = n-2;i >= 0;i--){
right = right*nums[i+1];
res[i] *= right;
}
return res;
}
}