303.区域和检索 - 数组不可变 (Easy)*

题目描述*

给定一个整数数组 nums，求出数组从索引 i 到 j (i ≤ j) 范围内元素的总和，包含 i, j 两点。

示例*

给定 nums = [-2, 0, 3, -5, 2, -1]，求和函数为 sumRange()

sumRange(0, 2) -> 1 sumRange(2, 5) -> -1 sumRange(0, 5) -> -3

说明*

你可以假设数组不可变。

会多次调用 sumRange 方法。

代码*

肯定不能每次都加，太蠢了。可以在构造函数初始化一个前缀和数组，之后求 i~j 的和直接 sum[j + 1] - sum[i] 就行了。

前缀和

class NumArray {
private:
    vector<int> sum;
public:
    NumArray(vector<int>& nums) {
        sum.push_back(0);
        for(int i = 0; i < nums.size(); i++) {
            sum.push_back(sum.back() + nums[i]);
        }
    }

    int sumRange(int i, int j) {
        return sum[j + 1] - sum[i];
    }
};

最后更新: July 23, 2022