238. 除自身以外数组的乘积
给你一个整数数组 nums,返回 数组 answer ,其中 answer[i] 等于 nums 中除 nums[i] 之外其余各元素的乘积 。
题目数据 保证 数组 nums之中任意元素的全部前缀元素和后缀的乘积都在 32 位 整数范围内。
请 不要使用除法,且在 O(n) 时间复杂度内完成此题。
示例 1:
输入: nums = [1,2,3,4]
输出: [24,12,8,6]
示例 2:
输入: nums = [-1,1,0,-3,3]
输出: [0,0,9,0,0]
提示:
2 <= nums.length <= 105L
-30 <= nums[i] <= 30
保证 数组 nums之中任意元素的全部前缀元素和后缀的乘积都在 32 位 整数范围内
解题思路
代码一
开两个数组,一个维护i的左侧乘积数组,一个维护i的右侧数组,那么所求答案就是 循环每个i的Ri*Li
class Solution {
public:
vector<int> productExceptSelf(vector<int>& nums) {
int len=nums.size();
vector<int> L(len,0),R(len,0);
vector<int> result(len);
L[0]=1;
R[len-1]=1;
for(int i=1;i<len;i++){
L[i]=nums[i-1]*L[i-1];
//cout<<L[i]<<' ';
}
//cout<<endl;
for(int i=len-2;i>=0;i--){
R[i]=R[i+1]*nums[i+1];
//cout<<R[i]<<" ";
}
//cout<<endl;
for(int i=0;i<len;i++){
result[i]=L[i]*R[i];
}
return result;
}
};
代码二
不开额外数组,将答案的result数组当作Li,r每次用 R *= nums[i]
去更新即可,这样能优化代码一的空间为O(1)
class Solution {
public:
vector<int> productExceptSelf(vector<int>& nums) {
int length = nums.size();
vector<int> answer(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;
}
};