题目 除自身以外的数组乘积
给你一个长度为 n 的整数数组 nums,其中 n > 1,返回输出数组 output ,其中 output[i] 等于 nums 中除 nums[i] 之外其余各元素的乘积。
题解 记录左右子列乘积
题目要求了时间复杂度为O(n)
,所以暴力解法无法生效,但是可以分别用数组记录在这个位置之前(之后)的乘积,最后让对应位置相乘,就可以达到效果,且满足条件。
class Solution {
public:
vector<int> productExceptSelf(vector<int>& nums) {
int n = nums.size();
vector<int> res(n, 0);
vector<int> Left(n, 0);
vector<int> Right(n, 0);
Left[0] = Right[n - 1] = 1;
for(int i = 1; i < n; ++i){
Left[i] = nums[i - 1] * Left[i - 1];
}
for(int j = n - 2; j >=0; --j){
Right[j] = Right[j + 1] * nums[j + 1];
}
for(int i = 0; i < n; ++i){
res[i] = Left[i] * Right[i];
}
return res;
}
};
时间复杂度O(n);
空间复杂度O(n);
改进
可以分别用Left和Right两个变量分别记录前缀积,同样满足条件,还使得空间复杂度将为了O(1);
class Solution {
public:
vector<int> productExceptSelf(vector<int>& nums) {
int n = nums.size();
vector<int> res(n, 1);
int Left = 1;
int Right = 1;
for(int i = 1; i < n; ++i){
Left = Left * nums[i - 1];
res[i] = Left;
}
for(int j = n - 2; j >= 0; --j){
Right = Right * nums[j + 1];
res[j] *= Right;
}
return res;
}
};