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
说明:

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

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/range-sum-query-mutable
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

class NumArray {
    int[] sum;
    int[] nums;
    int low;//修改的nums数组的最小索引
    boolean flag;//是否修改过
    public NumArray(int[] nums) {
        sum = new int[nums.length+1];
        for(int i=0;i<nums.length;i++){
            sum[i+1] = sum[i]+nums[i];
        }
        this.nums = nums;
        low = nums.length;
        flag = false;
    }
    
    public void update(int i, int val) {
        nums[i] = val;
        low = Math.min(low,i);
        flag = true;
    }
    
    public int sumRange(int i, int j) {
        if(flag){
            for(int k=low;k<nums.length;k++){
                sum[k+1] = sum[k]+nums[k];
            }
            low = nums.length;
            flag = false;
        }
        return sum[j+1]-sum[i];
    }
}

/**
 * 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);
 */
发布了250 篇原创文章 · 获赞 0 · 访问量 1280

猜你喜欢

转载自blog.csdn.net/qq_36198826/article/details/103781024