欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

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;
	}
};