LeetCode 303. Range Sum Query - Immutable

303. Range Sum Query - Immutable（区域和检索 - 数组不可变）

链接

https://leetcode-cn.com/problems/range-sum-query-immutable

题目

给定一个整数数组 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 方法。

思路

由于需要多次调用，建议面向答案编程，先建立一个sum数组，存放从开头到这个数的和，若要(i,j)那么就用sum[j]-sum[i-1]即可。

代码：

private int[] sum;

public void NumArray(int[] nums) {
  if(nums.length==0)
  {
    return;
  }
  sum = new int[nums.length];
  sum[0] = nums[0];
  for (int i = 1; i < nums.length; i++) {
    sum[i] += sum[i - 1] + nums[i];
  }
}

public int sumRange(int i, int j) {
  if (i == 0) {
    return sum[j];
  } else {
    return sum[j] - sum[i - 1];
  }
}