[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

解题思路

利用“前缀和”的思路即可，我们用动态规划记录前i的和，那么sum(i, j) = dp[j] - dp[i-1]

参考代码

// 每次调用函数计算一次都需要O(n)复杂度，所以我们借助缓存来保存区域和。
class NumArray {
public:
    NumArray(vector<int>& nums) {
        int sum = 0;
        for(int i = 0; i < nums.size(); i++){
            sum += nums[i];
            dp.push_back(sum);
        }
    }
    
    int sumRange(int i, int j) {
        if(i == 0)
            return dp[j];
        else
            return dp[j] - dp[i-1];
    }
    
private:
    vector<int> dp;
};

/**
 * Your NumArray object will be instantiated and called as such:
 * NumArray* obj = new NumArray(nums);
 * int param_1 = obj->sumRange(i,j);
 */