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];
}
}