问题描述
给你一个长度为 n 的整数数组 nums,其中 n > 1,返回输出数组 output ,其中 output[i] 等于 nums 中除 nums[i] 之外其余各元素的乘积。
示例:
输入: [1,2,3,4]
输出: [24,12,8,6]
解题报告
\qquad 因为题目不允许使用除法,所以可以暴力枚举,但是时间复杂度为 O ( N 2 ) O(N^2) O(N2),所以我们需要进行优化一下。在暴力计算的时候,很明显出现了很多重复计算,比如在计算除 nums[i] 之外其余各元素的乘积时,依次将左半部分相乘,然后再乘以右半部分;但是在计算 nums[i+1] 时,i+1的左半部分乘积和右半部分乘积与 i 的左半部分乘积和右半部分乘积存在很多交集。
\qquad 为了避免重复计算,我们首先将每个数的前缀积和后缀积计算出来,在这个计算过程中,我们可以使用转移方程,减少乘法操作的次数。
\qquad 最后我们只需要在遍历数组时,将位置 i 的前缀积和后缀积取出来相乘即可。总的时间复杂度为 O ( N ) O(N) O(N)。
对于暴力问题的优化,我们首先需要找出重复计算的操作,然后想办法能不能使用数组等数据结构将重复计算的数值提前存储起来,减少时间消耗。
\qquad
当然了,这道题的空间复杂度还有优化的空间。代码源自官方题解区。
鉴于前缀积和后缀积的特殊性,我们完全不需要重新开辟空间来存储前缀积和后缀积,完全可以在两遍遍历数组,将前缀积和后缀积依次存储在返回的结果数组中。
实现代码
- 未进行空间优化
class Solution {
public:
vector<int> productExceptSelf(vector<int>& nums) {
int n=nums.size();
vector<int>ans(n),Fmul(n+1,1),Amul(n+1,1);
for(int i=1;i<=n;i++){
Fmul[i]=Fmul[i-1]*nums[i-1];
}
for(int i=n-1;i>=0;i--){
Amul[i]=Amul[i+1]*nums[i];
}
for(int i=0;i<n;i++){
ans[i]=Fmul[i]*Amul[i+1];
}
return ans;
}
};
- 进行空间优化
class Solution {
public:
vector<int> productExceptSelf(vector<int>& nums) {
int R=1;
vector<int>ans(nums.size(),1);
for(int i=1;i<nums.size();i++){
ans[i]=ans[i-1]*nums[i-1];
}
for(int i=nums.size()-1;i>=0;i--){
ans[i]*=R;
R*=nums[i];
}
return ans;
}
};
参考资料
[1] Leetcode 238. 除自身以外数组的乘积
[2] 题解区:官方答案