编程问题:
给你一个长度为 n 的整数数组 nums,其中 n > 1,返回输出数组 output ,其中 output[i] 等于 nums 中除 nums[i] 之外其余各元素的乘积。
示例:
- 输入: [1,2,3,4]
输出: [24,12,8,6]
解法:
1.左右乘积
不必将所有数字的乘积除以给定索引处的数字得到相应的答案,而是利用索引左侧所有数字的乘积和右侧所有数字的乘积(即前缀与后缀)相乘得到答案。对于给定索引 i,我们将使用它左边所有数字的乘积乘以右边所有数字的乘积。
时间复杂度O(N)
空间复杂度O(N)
class Solution {
public:
vector<int> productExceptSelf(vector<int>& nums) {
int size = nums.size();
vector<int> res(size);
vector<int> L(size), R(size);
L[0] = 1;
for (int i = 1; i < size; i++)
{
L[i] = nums[i-1] * L[i-1];
}
R[size-1] = 1;
for (int i = size-2; i >= 0; i--)
{
R[i] = nums[i+1] * R[i+1];
}
for(int i = 0; i < size; i++)
{
res[i] = L[i] * R[i];
}
return res;
}
};
如果输出数组不算在空间复杂度内,那么可以将 L 或 R 数组用输出数组来计算。先把输出数组当作 L 数组来计算,然后再动态构造 R 数组得到结果。
时间复杂度O(N)
空间复杂度O(1)
class Solution {
public:
vector<int> productExceptSelf(vector<int>& nums) {
int size = nums.size();
vector<int> res(size);
res[0] = 1;
for (int i = 1; i < size; i++)
{
res[i] = nums[i-1] * res[i-1];
}
int R = 1;
for (int i = size - 1; i >= 0; i--)
{
res[i] *= R;
R *= nums[i];
}
return res;
}
};