LeetCode 307. 区域和检索 - 数组可修改（树状数组求解）

题目

给你一个数组 nums ，请你完成两类查询。

其中一类查询要求 更新 数组 nums 下标对应的值
另一类查询要求返回数组 nums 中索引 left 和索引 right 之间（ 包含 ）的nums元素的 和 ，其中 left <= right
实现 NumArray 类：

• NumArray(int[] nums) 用整数数组 nums 初始化对象
• void update(int index, int val) 将 nums[index] 的值 更新 为 val
• int sumRange(int left, int right)返回数组 nums 中索引 left 和索引 right 之间（ 包含 ）的nums元素的 和 （即，nums[left] +nums[left + 1], …, nums[right]）

提示：

• 1 <= nums.length <= 3 * 10^4
• -100 <= nums[i] <= 100
• 0 <= index < nums.length
• -100 <= val <= 100
• 0 <= left <= right < nums.length
• 调用 update 和 sumRange 方法次数不大于 3 * 10^4

题解

由于调用函数的次数和数组的长度都是10^4，所以这两个方法的时间复杂度均需要小于O(n)。
首先想到前缀和思路，前缀和的区间查询复杂度为O(1)，但是数组的修改和预处理复杂度为O(n),故不可使用。
所以采用树状数组模板，其单点修改和区间查询时间复杂度均为O(logn).

AC代码

class NumArray {
public:
    vector<int> tree;
    vector<int> numsarr;
    int n;
    NumArray(vector<int>& nums) {
        n = nums.size();
        numsarr = nums;
        tree.resize(n+1);
        for(int i=0;i<n;i++)
            update_increase(i+1,nums[i]);
    }
    int lowbit(int index){
        return index & (-index);
    }
    void update_increase(int index, int val){
        while(index <= n){
            tree[index] += val;
            index += lowbit(index);
        }
    }
    
    void update(int index, int val) {
        update_increase(index+1,val-numsarr[index]);
        numsarr[index]=val;
    }

    int query(int i) {
        int sum = 0;
        while (i > 0) {
            sum += tree[i];
            i -= lowbit(i);
        }
        return sum;
    }
    
    int sumRange(int left, int right) {
      
        return query(right+1)-query(left);
    }
};