[LeetCode] 303_区域和检索-数组不可变

[LeetCode] 303_区域和检索-数组不可变

题目要求

给定一个整数数组 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 方法。

题目分析

由于数组不可变并且需要多次读取，可以考虑将解或解的一部分事先进行存储。
在求解过程中，可以将数组遍历一遍，在遍历的过程中存储从0到当前位置所有数的解。
对于指定的区间[i，j]，可以用sum[j]-sum[i-1]来求得区间和。

注意点

当i==0时，i-1<0，返回sum[j]即可。

代码

class NumArray {
public:
    NumArray(vector<int> nums) {
        int sum=0;
        for(int i=0;i<nums.size();++i){
            sum+=nums[i];
            sums.push_back(sum);
        }
    }
    
    int sumRange(int i, int j) {
        if(i==0)return sums[j];
        return sums[j]-sums[i-1];
    }
private:
    vector<int>sums;
};