leetcode 238. Product of Array Except Self

题目概述

解题思路

这道题如果允许使用除法，那难度显然只有easy。。。既然这道题要求不能用除法，而且时间复杂度在O(n)，我们必然需要存储临时的结果用于提速度。不妨考虑一下，数组中的第 i 个乘积实际上是由它前面的 i - 1个的乘积，和后面的 len - i 个元素的乘积相乘得到的。我们可以先存储这两个数组，然后相乘，即可得结果。

我们还可以继续优化空间复杂度。思路就是，用最后返回的结果存储这个临时数组。

方法性能

O(N)时间复杂度， O(1)空间复杂度。

示例代码

class Solution {
public:
    vector<int> productExceptSelf(vector<int>& nums) 
    {
        int len = nums.size();
        vector<int> prefix(len, 1);
        for(int i = 1; i < len;++i)
        {
            prefix[i] = prefix[i - 1] * nums[i - 1];
        }
        
        int prod = 1;
        for(int i = len - 2; i >= 0; --i)
        {
            prod *= nums[i + 1];
            prefix[i] *= prod;
        }
        
        return prefix;
    }
};