Leetcode 307. 区域和检索 - 数组可修改（中等）

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

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

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

示例

Given nums = [1, 3, 5]
sumRange(0, 2) -> 9
update(1, 2)
sumRange(0, 2) -> 8

说明：

       数组仅可以在 update 函数下进行修改。
       你可以假设 update 函数与 sumRange 函数的调用次数是均匀分布的。

代码

class NumArray {
public:
    vector<int> tree;
    vector<int> Nums;
    int len;
    NumArray(vector<int>& nums) {
        len=nums.size();
        Nums=nums;
        tree=vector<int>(len+1,0);
        for(int i=0;i<len;i++){
            for(int j=i+1-lowbit(i+1);j<=i;j++){
                tree[i+1]+=nums[j];
            }
        }
    }
    
    void update(int i, int val) {
        for(int j=i;j<len;j+=lowbit(j+1)){
            tree[j+1]+=val-Nums[i];
        }
        Nums[i]=val;
    }
    
    int sumRange(int i, int j) {
        return getSum(j+1)-getSum(i);
    }
    
    int lowbit(int t){
        return t & (-t);
    }
    
    int getSum(int x){
        int ans=0;
        for(int i=x;i>0;i-=lowbit(i)){
            ans+=tree[i];
        }
        return ans;
    }
};

/**
 * 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);
 */