线段树 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 {
private:
    vector<int> nums;
    vector<int> sum;
public:
    NumArray(vector<int>& nums) {
        this->nums=nums;
        sum.resize(nums.size(),0);
        for(int i=0;i<nums.size();i++) sum[i]=(i>0?sum[i-1]:0)+nums[i];
    }
    
    void update(int i, int val) {
        for(int t=i;t<nums.size();t++)
            sum[t]=sum[t]-nums[i]+val;
        nums[i]=val;
    }
    
    int sumRange(int i, int j) {
        return sum[j]-(i>0?sum[i-1]:0);
    }
};

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

线段树

建立线段树
线段树为完全二叉树；
故建立数组大小为2*n，从n开始将nums放入数组中；
再从n-1~1对数组计算节点值：

        for(int i=n-1;i>0;i--)  //2*n+1  2*n+2
            Tree[i]=Tree[i*2]+Tree[i*2+1];

更新线段树
每次更新需要找到该叶节点位置：int index=n+i;；
进行赋值，后对改叶节点的根节点赋值——直到到最上层根节点；

输出两个节点之间的sum和
得到两个节点在树中的位置；
若left%2 == 1，此left为多出来的节点，sum+=Tree[left];left++;；
若right%2 == 0，此right为多出来的节点，sum+=Tree[right];right--;；
直到left与right相遇，加上相遇的这个节点值即可；

完整代码

class NumArray {
private:
    vector<int> Tree;
    int n;
    void BuildTree(vector<int> & nums){
        int n=nums.size();  
        for(int i=0,j=n;i<n;i++,j++)
            Tree[j]=nums[i];
        for(int i=n-1;i>0;i--)  //2*n+1  2*n+2
            Tree[i]=Tree[i*2]+Tree[i*2+1];
    }
public:
    NumArray(vector<int>& nums) {
        this->n=nums.size();
        Tree.resize(2*nums.size()+1,0);
        BuildTree(nums);
    }
    
    void update(int i, int val) {
        int index=n+i;
        int tmp=Tree[index];
        while(index>0){
            Tree[index]+=val-tmp;           
                index=index/2;
        }
    }
    
    int sumRange(int i, int j) {
        int left=n+i;
        int right=n+j;
        int sum=0;
        while(left<=right){
            if(left%2){
                sum+=Tree[left];
                left++;
            }
            if(right%2==0){
                sum+=Tree[right];
                right--;
            }
            left/=2;
            right/=2;
        }
        return sum;
    }
};

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