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

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

树状数组板子题，维护前缀和，控制时间复杂度

代码附上（注释）

树状数组知识点讲解看我这篇文章树状数组

class NumArray {
private:
    vector<int> tree; //树状数组
    vector<int> &nums;

    int lowBit(int x) {
        return x & -x;
    }

    void add(int index, int val) {
        while (index < tree.size()) {
            tree[index] += val;
            index += lowBit(index);//index是动态更新的
        }
    }

    int query(int index) { //index前缀和
        int sum = 0;
        while (index > 0) {
            sum += tree[index];
            index -= lowBit(index);//index是动态更新的
        }
        return sum;
    }

public:
    NumArray(vector<int>& nums) : tree(nums.size() + 1), nums(nums) {
        for (int i = 0; i < nums.size(); i++) {
            add(i + 1, nums[i]); //因为lowbit下标从1开始 树状数组初始化
        }
    }

    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);
    }
};

觉得文章有用，点个赞，点个收藏，支持一下