给你一个整数数组 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 位 整数范围内
进阶:你可以在 O(1) 的额外空间复杂度内完成这个题目吗?( 出于对空间复杂度分析的目的,输出数组不被视为额外空间。)
解题思路一:
这道题和 724. 寻找数组的中心下标 是类似的,只不过从之前的加法变成了乘法,具体的细节可以参考之前的笔记,同样还是用两个数组来分别记录前缀乘积和后缀乘积,最后由原数组来获取结果集!
其中 front 数组为前缀乘积数组,front[i] 表示从 [0, i - 1] 区间的元素乘积,所以可以很容易得到 front[i] = front[i - 1]*nums[i - 1],要注意的是遍历顺序要从左往右,并且 front[0] 要初始化为 1,因为这样子才不会影响后面的乘积!
而 back 数组为后缀乘积数组,back[i] 表示从 [i + 1, n - 1] 区间的元素乘积,很容易得到 back[i] = back[i + 1]*nums[i + 1],要注意的是遍历顺序是从后往前,并且 back[n - 1] 要初始化为 1。
class Solution {
public:
vector<int> productExceptSelf(vector<int>& nums) {
int n = nums.size();
vector<int> front(n, 1); // 前缀乘积数组,初始化为1
vector<int> back(n, 1); // 后缀乘积数组,初始化为1
// 前缀和处理
for(int i = 1; i < n; ++i)
front[i] = front[i - 1] * nums[i - 1];
for(int i = n - 2; i >= 0; --i)
back[i] = back[i + 1] * nums[i + 1];
// 获取结果
for(int i = 0; i < n; ++i)
nums[i] = front[i] * back[i];
return nums;
}
};解题思路二:
把高赞评论贴过来,给没有看懂的人方便理解,大佬绕路不用看了:
以前在做这个题的时候,老是搞不清楚边界的处理,看到别的题解说:先求下三角或者再求上三角之类的。今天正好借着【每日一题】系列,终于给搞懂了,不用管上三角或者下三角。见下图就知道了:
原数组: [1 2 3 4]
左部分的乘积: 1 1 1*2 1*2*3
右部分的乘积: 2*3*4 3*4 4 1
结果: 1*2*3*4 1*3*4 1*2*4 1*2*3*1 从上面的图可以看出,当前位置的结果就是它左部分的乘积再乘以它右部分的乘积。因此需要进行两次遍历,第一次遍历用于求左部分的乘积,第二次遍历在求右部分的乘积的同时,再将最后的计算结果一起求出来。
class Solution {
public:
vector<int> productExceptSelf(vector<int>& nums) {
int n = nums.size();
vector<int> answer(n, 1);
// answer[i] 表示索引 i 左侧所有元素的乘积
// 因为下标为0的元素左侧没有元素,所以 answer[0] = 1
for(int i = 1; i < n; ++i)
answer[i] = answer[i - 1] * nums[i - 1];
// 为了O(1)空间复杂度,这里使用R来遍历获取每次i右侧的乘积即可
int R = 1;
for(int i = n - 1; i >= 0; --i)
{
answer[i] = answer[i] * R;
R *= nums[i];
}
return answer;
}
};