力扣第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]

分析

题目中明确说明不能使用除法 所以我们可以把除自身以外数组的乘积看成该元素的前面所有元素的乘积（前缀积）乘与后面所有的元素的乘积（后缀积）就为除自身以外的乘积

代码如下：

class Solution {
public:
    vector<int> productExceptSelf(vector<int>& nums) {
        // 创建一个结果数组，初始值为 1，大小与输入数组相同
        vector<int> res(nums.size(), 1);
        int prefix = 1; // 前缀积初始值为 1
        // 计算每个元素的左侧所有元素的乘积（即前缀积）
        for (int i = 0; i < nums.size(); ++i) {
            res[i] = prefix; // 将当前位置的前缀积存入结果数组中
            prefix = prefix * nums[i]; // 更新前缀积，乘以当前元素
        }
        
        int postfix = 1; // 后缀积初始值为 1
        
        // 计算每个元素的右侧所有元素的乘积（即后缀积），并与之前的结果相乘
        for (int j = nums.size() - 1; j >= 0; --j) {
            res[j] = res[j] * postfix; // 将当前位置的后缀积乘到之前的结果上
            postfix = postfix * nums[j]; // 更新后缀积，乘以当前元素
        }
        // 返回最终的结果数组
        return res;
    }
};

解释注释

1.vector res(nums.size(), 1);

创建一个大小与输入数组 nums 相同的结果数组 res，并将所有元素初始化为 1，这是因为乘积的初始值为 1。

2.int prefix = 1;

初始化变量 prefix，用于存储当前元素左侧所有元素的乘积的累积值，初始值为 1。

3.第一个循环 (for (int i = 0; i < nums.size(); ++i))：计算前缀积

res[i] = prefix;
将当前位置 i 的前缀积存入结果数组 res 中。
prefix = prefix * nums[i];
更新 prefix，将其乘以当前元素 nums[i]，得到新的前缀积。

4.int postfix = 1;

初始化变量 postfix，用于存储当前元素右侧所有元素的乘积的累积值，初始值为 1。

5.第二个循环 (for (int j = nums.size() - 1; j >= 0; --j))：计算后缀积并与前缀积相乘

res[j] = res[j] * postfix;
将当前位置 j 的后缀积乘到之前计算得到的结果上，实现左右乘积的合并。
postfix = postfix * nums[j];
更新 postfix，将其乘以当前元素 nums[j]，得到新的后缀积。

时间复杂度

这个方法的时间复杂度是 O(n)，其中 n 是数组 nums 的大小。算法进行了两次线性遍历，每次遍历都是 O(n) 的时间复杂度，因此总体时间复杂度是 O(n)。杂度。