leetcode303. 区域和检索 - 数组不可变

传送门

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

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

sumRange(0, 2) ->  1
sumRange(2, 5) -> -1
sumRange(0, 5) -> -3
注意： 会多次调用 sumRange 方法。

记录前n项和，用sum(j) - sum(i) 得到 i到j的和

原始方法：要特判nums为null的情况

class NumArray {
    int[] dpSum;
    public NumArray(int[] nums) {
        if (nums.length == 0) dpSum = null;
        else {
            dpSum = new int[nums.length];
            dpSum[0] = nums[0];
            for (int i = 1; i < nums.length; ++i) {
                dpSum[i] = dpSum[i - 1] + nums[i];
            }
        }
    }
    
    public int sumRange(int i, int j) {
        if (dpSum == null) return 0;
        if (i == 0) return dpSum[j];
        return dpSum[j] -dpSum[i - 1];
    }
}

优化方法：在dpSum[0]插入了一个虚拟 0 ,作为 sum 数组中的第一个元素。

这个技巧可以避免像上面一样进行额外nums=null的条件检查

	int[] dpSum;// dpSum[i] 代表 [0, i-1)的数字和
    public NumArray(int[] nums) {
        dpSum = new int[nums.length + 1];
        for (int i = 0; i < nums.length; ++i) {
            dpSum[i + 1] = dpSum[i] + nums[i];
        }
    }
    
    public int sumRange(int i, int j) {
        return dpSum[j + 1] - dpSum[i] ;
    }