LeetCode-238-除自身以外数组的乘积
思路
想要计算的东西很简单,一般可以考虑计算全部乘积,然后除以当前数就可以,但是如果存在0,那么全部都将失去意义,会导致全部乘积变为0,因此考虑其他方法
左右乘积数组法
可以计算每个元素的左右乘积,然后左右相乘即可
左边:
- 第1个元素左边没有元素,因此L[0]=1
- 第i个元素,L[i]=L[i-1]*nums[i-1]
右边:
- 第len-1个元素右边没有元素,因此R[len-1]=1
- 第i个元素,R[i]=R[i+1]*nums[i+1]
res[i]=L[i]*R[i]
代码
//左右乘积数组法
class Solution {
public int[] productExceptSelf(int[] nums) {
int len=nums.length;
int []L=new int[len];
int []R=new int[len];
int []res=new int[len];
L[0]=1;
for(int i=1;i<len;i++){
L[i]=L[i-1]*nums[i-1];
}
R[len-1]=1;
for(int i=len-2;i>=0;i--){
R[i]=R[i+1]*nums[i+1];
}
for(int i=0;i<len;i++){
res[i]=L[i]*R[i];
}
return res;
}
}
//优化空间复杂度为1,
class Solution {
public int[] productExceptSelf(int[] nums) {
int len=nums.length;
int []res=new int[len];
res[0]=1;
for(int i=1;i<len;i++){
res[i]=res[i-1]*nums[i-1];
}
int R=1;//R表示右数组,可以用滚动数组实现,因此只维护一个变量即可
for(int i=len-1;i>=0;i--){
res[i]*=R;
R*=nums[i];
}
return res;
}
}