leetcode- 区域和检索 - 数组可修改

给你一个数组 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:

输入:
[“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

提示:

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

java代码:
class NumArray {
    int[] sums;
    int[] nums;
    int n;
    public NumArray(int[] nums) {
        n = nums.length;
        sums = new int[n+1];
        this.nums = nums;
        for(int i = 0; i < n; i++){
            add(i,nums[i]);
        }
    }

    private void add(int index, int item){
        int i = index+1;
        while(i<=n){
            sums[i]+=item;
            i += last(i);
        }
    }

    private int get(int index){
        int i = index+1;
        int ans = 0;
        while(i>0){
            ans += sums[i];
            i -= last(i);
        }
        return ans;
    }

    private int last(int a){
        return a & (-a);
    }
    
    public void update(int index, int val) {
        int add = val-nums[index];
        nums[index] = val;
        add(index,add);
    }
    
    public int sumRange(int left, int right) {
        return get(right)-get(left-1);
    }
}

/**
 * Your NumArray object will be instantiated and called as such:
 * NumArray obj = new NumArray(nums);
 * obj.update(index,val);
 * int param_2 = obj.sumRange(left,right);
 */

你可能感兴趣的:(leetcode-数组,LeetCode,leetcode)