238. 除自身以外数组的乘积
程序员文章站
2022-07-14 18:06:11
...
给定长度为 n 的整数数组 nums,其中 n > 1,返回输出数组 output ,其中 output[i] 等于 nums 中除 nums[i] 之外其余各元素的乘积。
示例:
输入: [1,2,3,4]
输出: [24,12,8,6]
说明: 请不要使用除法,且在 O(n) 时间复杂度内完成此题。
思路:
注意不能使用除法,不然我们可以根据总积除以当前数,当时考虑好0的情况。
现在用左右乘积的方法,从左到右遍历,获得每个元素左边的总乘积,再从右到左遍历,获得每个元素右边的总乘积。下面的代码实现这个思路,但是空间复杂度不是1
class Solution {
public:
vector<int> productExceptSelf(vector<int>& nums)
{
vector<int> output;
vector<int> leftProduct;
vector<int> rightProduct;
int leftTemp = 1, rightTemp = 1;
for (int i = 0; i < nums.size(); ++i)
{
leftProduct.push_back(leftTemp);
leftTemp *= nums[i];
}
for (int i = nums.size() - 1; i >= 0; --i)//注意rightProduct的角标顺序是反着的
{
rightProduct.push_back(rightTemp);
rightTemp *= nums[i];
}
for (int i = 0; i < nums.size(); ++i)
{
output.push_back(leftProduct[i] * rightProduct[nums.size()-i-1]);
}
return output;
}
};
思路2:
对上面的思路进行改进,直接用output做为左边总乘积
class Solution {
public:
vector<int> productExceptSelf(vector<int>& nums)
{
vector<int> output;
int leftTemp = 1, rightTemp = 1;
for (int i = 0; i < nums.size(); ++i)
{
output.push_back(leftTemp);
leftTemp *= nums[i];
}
for (int i = nums.size()-1; i >= 0; --i)
{
output[i] *= rightTemp;
rightTemp *= nums[i];
}
return output;
}
};