每日OJ题_算法_前缀和④_力扣238. 除自身以外数组的乘积

目录

力扣238. 除自身以外数组的乘积

解析代码

---

力扣238. 除自身以外数组的乘积

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

难度 中等

给你一个整数数组 nums，返回 数组 answer ，其中 answer[i] 等于 nums 中除 nums[i] 之外其余各元素的乘积 。

题目数据 保证 数组 nums之中任意元素的全部前缀元素和后缀的乘积都在  32 位 整数范围内。

请 不要使用除法，且在 O(n) 时间复杂度内完成此题。

示例 1:

输入: nums = [1,2,3,4]
输出: [24,12,8,6]

示例 2:

输入: nums = [-1,1,0,-3,3]
输出: [0,0,9,0,0]

提示：

• 2 <= nums.length <= 10^5
• -30 <= nums[i] <= 30
• 保证 数组 nums之中任意元素的全部前缀元素和后缀的乘积都在  32 位 整数范围内

进阶：你可以在 O(1) 的额外空间复杂度内完成这个题目吗？（ 出于对空间复杂度分析的目的，输出数组 不被视为 额外空间。）

class Solution {
public:
    vector productExceptSelf(vector& nums) {

    }
};

---

解析代码

题目的提示就差直接告诉你用前缀和思想了，类似力扣724的代码。

class Solution {
public:
    vector productExceptSelf(vector& nums) {
        int n = nums.size();
        vector arr1(n, 1), arr2(n, 1), ret(n);
        for(int left = 1, right = n-2; left < n; ++left, --right)
        {
            arr1[left] = arr1[left-1] * nums[left-1]; // 前缀积
            arr2[right] = arr2[right+1] * nums[right+1]; // 后缀积
        }
        for(int i = 0; i < n; ++i)
        {
            ret[i] = arr1[i] * arr2[i];
        }
        return ret;
    }
};