Leetcode|前缀和(双向前缀积)|剑指 Offer 66. 构建乘积数组

最新推荐文章于 2023-08-27 15:54:18 发布

原创最新推荐文章于 2023-08-27 15:54:18 发布 · 305 阅读

0 ·

CC 4.0 BY-SA版权

Leetcode其他经典题专栏收录该内容

61 篇文章

订阅专栏

本文介绍了一种使用双向前缀积的方法来解决特定数组问题的C++实现。该算法通过两次遍历输入数组，一次从左到右计算前缀积，另一次从右到左计算后缀积，最终得到每个元素对应位置的乘积结果。

在这里插入图片描述

1 前缀和（双向前缀积）

class Solution {
public:
    vector<int> constructArr(vector<int>& a) {
        int size = a.size();
        vector<int> pre(size + 1, 1);
        vector<int> last(size + 1, 1);
        vector<int> res(size, 1);
        // 前缀积（从左至右乘积）
        for (int i = 0; i < size; i++)
            pre[i + 1] = pre[i] * a[i];
        // 后缀积
        for (int i = size - 1; i >= 0; i--)
            last[i] = last[i + 1] * a[i];
        // 计算结果
        for (int i = 0; i < size; i++) 
            res[i] = pre[i] * last[i + 1];
        return res;
    }
};