leetcode刷题. 238. 除自身以外数组的乘积

给你一个长度为 n 的整数数组 nums,其中 n > 1,返回输出数组 output ,其中 output[i] 等于 nums 中除 nums[i] 之外其余各元素的乘积。

leetcode刷题. 238. 除自身以外数组的乘积_第1张图片 

空间换时间。

    vector productExceptSelf(vector& nums) {

        vector output;

        vector arr1(nums.size(), 1);
        vector arr2(nums.size(), 1);

        for(int i = 1; i < nums.size(); i++)
            arr1[i] = arr1[i - 1] * nums[i - 1];

        for(int i = nums.size() - 2; i >= 0; i--)
            arr2[i] = arr2[i + 1] * nums[i + 1];

        for(int j = 0; j < nums.size(); j++) 
            output.push_back(arr1[j] * arr2[j]);
        
        return output;
    }

时间复杂度:o(n)

空间复杂度:o(n)

优化版本:这里输出数组不计。

    vector productExceptSelf(vector& nums) {

        vector output(nums.size(), 1);
        int tmpRight = 1;

        for(int i = 1; i < nums.size(); i++)
            output[i] = output[i - 1] * nums[i - 1];
        
        for(int j = nums.size() - 1; j >= 0; j--) {

            output[j] *= tmpRight;
            tmpRight *= nums[j]; 
        }
        return output;
    }

空间复杂度:O(1)

你可能感兴趣的:(【leetcode刷题】,【算法】)