Leetcode 238 除自身以外数组的乘积 前后缀分解

我们发现，对于每个数字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;
    }
};