leetcode：303. 区域和检索 - 数组不可变

题目来源

303. 区域和检索 - 数组不可变

题目描述

class NumArray {
public:
    NumArray(vector<int>& nums) {

    }
    
    int sumRange(int left, int right) {

    }
};

题目解析

sumRange 函数需要计算并返回一个索引区间之内的元素和，没学过前缀和的人可能写出如下代码：

class NumArray {
    vector<int> nums;
public:
    NumArray(vector<int>& nums) {
        this->nums = nums;
    }

    int sumRange(int left, int right) {
        int res = 0;
        for (int i = left; i <= right; ++i) {
            res += nums[i];
        }
        return res;
    }
};

这样，可以达到效果，但是效率很差，因为 sumRange 方法会被频繁调用，而它的时间复杂度是 O(N)，其中 N 代表 nums 数组的长度。

这道题的最优解法是使用前缀和技巧，将 sumRange 函数的时间复杂度降为 O(1)，说白了就是不要在 sumRange 里面用 for 循环，咋整？

class NumArray {
    vector<int> preSum{};
public:
    NumArray(vector<int>& nums) {
        preSum.resize(nums.size() + 1);
        // 计算 nums 的累加和
        for (int i = 1; i < preSum.size(); ++i) {
            preSum[i] = preSum[i - 1] + nums[i - 1];
        }
    }

    /* 查询闭区间 [left, right] 的累加和 */
    int sumRange(int left, int right) {
        return preSum[right + 1] - preSum[left];
    }
};

核心思路是我们 申请一个新的数组 preSum 出来，preSum[i] 记录 nums[0…i-1] 的累加和，看图 10 = 3 + 5 + 2：

看这个 preSum 数组，如果我想求索引区间 [1, 4] 内的所有元素之和，就可以通过 preSum[5] - preSum[1] 得出。

这样，sumRange 函数仅仅需要做一次减法运算，避免了每次进行 for 循环调用，最坏时间复杂度为常数 O(1)。

前缀和适用场景：原始数组不会被修改的情况下，频繁查询某个区间的累加和