1、题目描述
给你一个长度为 n 的整数数组 nums,其中 n > 1,返回输出数组 output ,其中 output[i] 等于 nums 中除 nums[i] 之外其余各元素的乘积。
提示:题目数据保证数组之中任意元素的全部前缀元素和后缀(甚至是整个数组)的乘积都在 32 位整数范围内。
说明: 请不要使用除法,且在 O(n) 时间复杂度内完成此题。
进阶:
你可以在常数空间复杂度内完成这个题目吗?( 出于对空间复杂度分析的目的,输出数组不被视为额外空间。)
2、示例
输入: [1,2,3,4]
输出: [24,12,8,6]
3、题解
基本思想:左右乘积列表,L[i]是i左侧所有数字的乘积,R[i]是i右侧所有数字的乘积,索引i处的值为L[i-1]*R[i+1],时间复杂度O(n)空间复杂度O(n)。
#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
class Solution {
public:
vector<int> productExceptSelf(vector<int>& nums) {
//基本思想:左右乘积列表,L[i]是i左侧所有数字的乘积,R[i]是i右侧所有数字的乘积
//索引i处的值为L[i-1]*R[i+1],时间复杂度O(n)空间复杂度O(n)
vector<int> res;
vector<int> L(nums.size(), 1);
vector<int> R(nums.size(), 1);
L[0] = nums[0];
for (int i = 1; i < nums.size(); i++)
L[i] = nums[i] * L[i - 1];
R[nums.size() - 1] = nums[nums.size() - 1];
for (int i = nums.size() - 2; i >= 0; i--)
R[i] = nums[i] * R[i + 1];
res.push_back(R[1]);
for (int i = 1; i < nums.size() - 1; i++)
res.push_back(L[i - 1] * R[i + 1]);
res.push_back(L[nums.size() - 2]);
return res;
}
};
class Solution1 {
public:
vector<int> productExceptSelf(vector<int>& nums) {
//基本思想:左右乘积列表,利用nums作为左乘积列表,变量R作为右乘积列表,时间复杂度O(n)空间复杂度O(1)
vector<int> res(nums.size(), 1);
res[0] = nums[0];
for (int i = 1; i < nums.size(); i++)
res[i] = nums[i] * res[i - 1];
int R = 1;
for (int i = nums.size() - 1; i >= 1; i--)
{
res[i] = res[i - 1] * R;
R *= nums[i];
}
res[0] = R;
return res;
}
};
int main()
{
Solution solute;
vector<int> nums = { 1,2,3,4 };
vector<int> res = solute.productExceptSelf(nums);
for_each(res.begin(), res.end(), [](const auto v) {cout << v << endl; });
return 0;
}