【LeetCode】307. 区域和检索 - 数组可修改

1.题目

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

update(i, val) 函数可以通过将下标为 i 的数值更新为 val，从而对数列进行修改。

点击跳转原题

2.思路

在上题的基础上，增加update函数，用于更新数值；
先求出nums[i]=num[i+1]-num[i];
求出val与nums[i]的差值den；
在后续数组num中依次增加den即可。

3.代码

class NumArray {
public:
    vector<int>num;
    NumArray(vector<int> nums) {
        num.resize(nums.size()+1,0);
        num[0]=0;
        for(int i=0;i<nums.size();i++){
            num[i+1]=nums[i]+num[i];
        }
    }
    
    void update(int i, int val) {
        int den=val-(num[i+1]-num[i]);
        cout<<den<<endl;
        for(i++;i<num.size();i++)
            num[i]+=den;
    }
    
    int sumRange(int i, int j) {
        return num[j+1]-num[i];
    }
};

/**
 * Your NumArray object will be instantiated and called as such:
 * NumArray obj = new NumArray(nums);
 * obj.update(i,val);
 * int param_2 = obj.sumRange(i,j);
 */