LeetCode：307. 区域和检索 – 数组可修改（树状数组 C++）

本文介绍: 如果没有更新，用前缀和就行，但是此题数组会改变，如果每次都求一次前缀和一定超时，所以考虑用树状数组。

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

实现 NumArray 类：

输入：
["NumArray", "sumRange", "update", "sumRange"]
[[[1, 3, 5]], [0, 2], [1, 2], [0, 2]]
输出：
[null, 9, null, 8]

解释：
NumArray numArray = new NumArray([1, 3, 5]);
numArray.sumRange(0, 2); // 返回 1 + 3 + 5 = 9
numArray.update(1, 2);   // nums = [1,2,5]
numArray.sumRange(0, 2); // 返回 1 + 2 + 5 = 8

class NumArray {
public:
    vector<int> tr = vector<int>(1000010);
    int lowbit(int x) {
        return x &amp; -x;
    }

    int query(int x) {
        int res = 0;
        for (int i = x; i > 0; i -= lowbit(i)) res += tr[i];
        return res;
    }

    void add(int x, int u) {
        for (int i = x; i <= n; i += lowbit(i)) tr[i] += u;
    }



    vector<int> nums;
    int n;
    NumArray(vector<int>&amp; nums) {
        n = nums.size();
        this->nums = nums;
        // 初始化 树状数组
        tr.resize(n + 1, 0);
        for (int i = 0; i < n; i++) add(i + 1, nums[i]);
    }
    
    void update(int index, int val) {

        add(index + 1, val - nums[index]);
        nums[index] = val;
    }
    
    int sumRange(int left, int right) {
        return query(right + 1) - query(left);
    }
};