238. 除自身以外数组的乘积https://leetcode.cn/problems/product-of-array-except-self/
给你一个整数数组 nums
,返回 数组 answer
,其中 answer[i]
等于 nums
中除 nums[i]
之外其余各元素的乘积 。
题目数据 保证 数组 nums
之中任意元素的全部前缀元素和后缀的乘积都在 32 位 整数范围内。
请 不要使用除法,且在 O(n)
时间复杂度内完成此题。
示例 1:
输入: nums =[1,2,3,4]
输出:[24,12,8,6]
示例 2:
输入: nums = [-1,1,0,-3,3] 输出: [0,0,9,0,0]
提示:
2 <= nums.length <= 105
-30 <= nums[i] <= 30
- 保证 数组
nums
之中任意元素的全部前缀元素和后缀的乘积都在 32 位 整数范围内
对于这道题,它在力扣中是归类在前缀和算法中的题目。这道题解法很多,我们不妨利用前缀和的思想 思考一下解题方法。
思路
利用前缀和的思想,分别从前 后两个方向计算各个数累乘结果并记录,这样如果我们想要在总的乘积中除以一个数nums[i],那就等价于nums[0]nums[1]...nums[i-1]和 nums[i+1]nums[i+2]...nums[n-1]的乘积,我们只需要对这些乘积有所记录,那么想要用的时候直接用就可以了。
解题过程
下面的代码中,我定义了frontt数组和afterr数组来记录累乘结果。
对于两个数组的解释,示例如下:
frontt[i]=nums[0] * nums[1] * ... * nums[i-1];
afterr[i]=nums[n-1] * nums[n-2] * ... * nums[i+1];
现在我们来求解去除nums[i]对应的乘积结果:
应有如下等式:ans[i]=frontt[i] * afterr[n-i-2];
n-i-2其实是n-1-(i+1);
复杂度
时间复杂度: O(n)
空间复杂度: O(n)
Code
C++
class Solution {
public:
vector<int> productExceptSelf(vector<int>& nums) {
int n=nums.size();
if(n==2) return{nums[1],nums[0]};//判断边界条件
vector<int> frontt(n);
vector<int> afterr(n);
frontt[0]=nums[0],afterr[0]=nums[n-1];
//分别计算前缀和后缀的乘积
for(int i=1;i<n;i++){
frontt[i]=frontt[i-1]*nums[i];
afterr[i]=afterr[i-1]*nums[n-i-1];
}
vector<int> ans(n);//去除nums[i]的乘积结果
ans[0]=afterr[n-2];ans[n-1]=frontt[n-2];
for(int i=1;i<n-1;i++) ans[i]=frontt[i-1]*afterr[n-i-2];
return ans;
}
};