给你一个整数数组 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]
思路:
分两次循环遍历数组,第一次先用ans[i]记录nums[i]左边所有元素的乘积,第二次用r在线记录右边所有元素的乘积,左右相乘得到结果并更新到ans中,此时ans[i]已无用。
class Solution {
public:
vector<int> productExceptSelf(vector<int>& nums) {
int n = nums.size();
vector<int> ans(n, 1);
int r = 1;
for (int i = 1; i < n; i++)
{
ans[i] = ans[i-1] * nums[i-1];
}
for (int j = n-2; j >= 0; j--)
{
r *= nums[j+1];
ans[j] *= r;
}
return ans;
}
};