区域和检索-数组不可变

给定一个整数数组 nums，求出数组从索引 i 到 j (i ≤ j) 范围内元素的总和，包含 i, j 两点。

示例：

1
2
3
4
5
给定 nums = [-2, 0, 3, -5, 2, -1]，求和函数为 sumRange()

sumRange(0, 2) -> 1
sumRange(2, 5) -> -1
sumRange(0, 5) -> -3

说明:

  1. 你可以假设数组不可变。
  2. 会多次调用 sumRange 方法。

代码

由于会多次调用sumRange方法，所以不应该在获取时采用for循环计算结果值，而是使用一个数组保存前n个元素的和，需要获取范围内元素总和时，直接用前j个元素的和-前i个元素的和

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
class NumArray {
    //定义一个保存前n个元素和的数组
    private int[] sums;

    public NumArray(int[] nums) {
        //如果nums为空或者长度为0,直接结束构造器方法
        if(nums==null||nums.length==0)return;
        sums=new int[nums.length];
        sums[0]=nums[0];
        for(int i=1;i<nums.length;i++){
            //前i个元素的和等于前一个元素加上第i个元素的值
            sums[i]=sums[i-1]+nums[i];
        }
    }
    
    public int sumRange(int i, int j) {
        //如果i==0，则不需要减
        if(i==0)return sums[j];
        //否则减去前i-1个元素的和
        return sums[j]-sums[i-1];
    }
}