题目描述
给你一个整数数组 nums,返回 数组 answer ,其中 answer[i] 等于 nums 中除 nums[i] 之外其余各元素的乘积 。
题目数据 保证 数组 nums之中任意元素的全部前缀元素和后缀的乘积都在 32 位 整数范围内。
请 不要使用除法,且在 O(n) 时间复杂度内完成此题。
出处
思路
前缀积x后缀积。
代码
class Solution {
public:
vector<int> productExceptSelf(vector<int>& nums) {
int n = nums.size();
vector<int> answer(n);
int* left=new int[n+1];
int* right=new int[n+1];
left[0]=1;
right[n]=1;
for(int i=1;i<=n;i++){
left[i]=left[i-1]*nums[i-1];
right[n-i]=right[n+1-i]*nums[n-i];
}
for(int i=0;i<n;i++){
answer[i]=left[i]*right[i+1];
}
return answer;
}
};
//空间O(1)优化解
class Solution {
public:
vector<int> productExceptSelf(vector<int>& nums) {
int n = nums.size();
vector<int> answer(n);
int left = 1, right = 1;
for (int i = 0; i < n; i++) {
answer[i] = left;
left *= nums[i];
}
for (int j = n - 1; j >= 0; j--) {
answer[j] *= right;
right *= nums[j];
}
return answer;
}
};