我们发现,对于每个数字nums[i], 对应的res[i]可以分解为i的前缀和i的后缀。如果我们首先预处理一下所有前缀乘积和所有后缀乘积,就可以在O(N)的时间完成解答。
但此时我们开了三个额外的数组
class Solution {
public:
vector<int> productExceptSelf(vector<int>& nums) {
int n = nums.size();
vector<int> pre(n,1);
vector<int> post(n,1);
vector<int> res(n);
for(int i=1;i<n;i++) pre[i] = pre[i-1]*nums[i-1];
for(int i=n-2;i>=0;i--) post[i] = post[i+1]*nums[i+1];
for(int i=0;i<n;i++) res[i] = pre[i]*post[i];
return res;
}
};
是否可以优化空间复杂度,是可以的,我们先把答案处理为后缀乘积,然后再向后遍历的时候,用临时变量记录下前缀乘积,这样就最终实现了除答案数组外O(1)的空间。
class Solution {
public:
vector<int> productExceptSelf(vector<int>& nums) {
int n = nums.size();
vector<int> res(n,1);
for(int i=n-2;i>=0;i--) res[i] = res[i+1]*nums[i+1];
int pre = 1;
for(int i=0;i<n;i++){
res[i]*=pre;
pre*=nums[i];
}
return res;
}
};