Leetcode 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

说明:

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

class NumArray {
    int[] tree;
    int n;
    public NumArray(int[] nums){
        if(nums.length>0){
            n=nums.length;
            tree=new int[n*2];
            buildTree(nums);
        }
    }
    private void buildTree(int[] nums){
        for(int i=n,j=0;i<2*n;i++,j++){
            tree[i]=nums[j];
        }
        for(int i=n-1;i>0;--i){
            tree[i]=tree[i*2]+tree[i*2+1];
        }
    }

    void update(int pos,int val){
        pos+=n;
        tree[pos]=val;
        while(pos>0){
            int left=pos;
            int right=pos;
            if(pos%2==0){
                right=pos+1;
            }else{
                left=pos-1;
            }
            tree[pos/2]=tree[left]+tree[right];
            pos/=2;
        }
    }

    public int sumRange(int l,int r){
        l+=n;
        r+=n;
        int sum=0;
        while(l<=r){
            if((l%2)==1){
                sum+=tree[l];
                l++;
            }
            if((r%2)==0){
                sum+=tree[r];
                r--;
            }
            l/=2;
            r/=2;
        }
        return sum;
    }
}