区域和检索 - 数组不可变

254 阅读1分钟

给定一个整数数组

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

说明:

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

class NumArray {

    //创建预处理数组,存储前缀和
    private int[] prefixSum;

    public NumArray(int[] nums) {
        //数组多申请一个长度
        prefixSum = new int[nums.length + 1];
        int sum = 0;
        //记录 prefixSum[0] = 0,防止数组越界 因为要包含第i位的元素
        prefixSum[0] = 0;
        for (int i = 0; i < nums.length; i++) {
            sum += nums[i];
            prefixSum[i + 1] = sum;
        }
    }

    public int sumRange(int i, int j) {
        return prefixSum[j + 1] - prefixSum[i];
    }
}