给你一个长度为 n 的整数数组 nums,其中 n > 1,返回输出数组 output ,其中 output[i] 等于 nums 中除 nums[i] 之外其余各元素的乘积。
示例:
输入: [1,2,3,4]
输出: [24,12,8,6]
提示:题目数据保证数组之中任意元素的全部前缀元素和后缀(甚至是整个数组)的乘积都在 32 位整数范围内。
说明: 请不要使用除法,且在 O(n) 时间复杂度内完成此题。
进阶:你可以在常数空间复杂度内完成这个题目吗?( 出于对空间复杂度分析的目的,输出数组不被视为额外空间。)
思路:看到这道题第一个想到的就是把所有数字乘起来在除数组本身的吧,但是这样有0的话会出错,而且题目本身要求不能用除法,应该换个思路。
题目每个数的结果是前缀积与后缀积的乘积,而很容易得知从左往右的前缀数、后缀数规律分别为递增和递减,换言之我们可以分别从从左和从右遍历来得到前缀积和后缀积(递乘?),然后再相乘得到结果。这样的话得遍历两次,第一次遍历的结果需要用一个长度为n的数组来储存(先前缀或者后缀都可),题目的进阶说明为输出数组不被视为额外空间,因此第一次遍历的结果用输出数组存储即可,我选择先从左往右得到前缀积,代码如下
class Solution {
public:
vector<int> productExceptSelf(vector<int>& nums) {
int len = nums.size();
vector<int> ans(len);
//后缀积初始化为1
int R = 1;
//前缀积初始化为1
ans[0]=1;
//从左往右遍历得到前缀积
for(int i=1;i<len;i++){
ans[i]=nums[i-1]*ans[i-1];
}
//从右往左遍历得到后缀积并与前缀积相乘得到结果
for(int i=len-1;i>=0;i--){
ans[i]*=R;
R*=nums[i];
}
return ans;
}
};
运行结果: