LeetCode之Product of Array Except Self

最新推荐文章于 2024-10-03 07:19:37 发布

原创最新推荐文章于 2024-10-03 07:19:37 发布 · 332 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#Leetcode

Leetcode 专栏收录该内容

168 篇文章

订阅专栏

本文介绍了一种O(n)时间复杂度的算法来解决计算数组中除当前元素外所有元素乘积的问题。该方法通过使用两个数组sums1和sums2来分别从前向后和从后向前计算累积乘积，然后将两者结合得到最终结果。此外还提供了一个常数空间复杂度的优化方案。

/*时间复杂度为O(n)的解法：
开两个数组sums1[n]、sums2[n]，开数组res[n]存结果。
1.sums1元素值为：sums1[i] = num[0]*nums[1]*...*nums[i];
2.sums2元素值为：sums2[i] = num[n-1]*nums[n-2]*...*nums[i];(n为数组nums长度)；
3.则结果res[i] = sums1[i-1] * sums2[i+1];
常数空间的解法：
为了减少空间复杂度，对上述方法进行如下改进：
1.用res[n]代替sums1做临时存储；
2.用一个int值sum_tmp记录从后往前的开始相乘的值；
3.res[i]等于res[i-1] * tmp。
*/
class Solution {
public:
    vector<int> productExceptSelf(vector<int>& nums) {
        if(nums.size() <= 1) return nums;
        vector<int> res(nums);
        for(int i = 1; i < nums.size(); ++i) res[i] = res[i-1] * nums[i];
        int tmp(1);
        for(int i = nums.size()-1; i >= 0; --i){
            if(i == 0) res[i] = tmp;
            else{
                res[i] = res[i-1] * tmp;
                tmp *= nums[i];          
            }
        }
        return res;
    }
};