题目:
给定一个整数数组 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 方法。
思路:
最简单的思路就是每次调用sumRange都根据i和j来对数组切片,求和,但是这样在多次调用sumRange时,会重复计算一些数据,浪费时间。所以最快速的方法,应该求所有[0,i]区间的和,然后[i,j] = [0,j] - [0, i- 1],这样只用做一次减法运算。
代码:
class NumArray {
private int[] sums;
public NumArray(int[] nums) {
int len = nums.length;
sums = new int[nums.length+1];
if (nums.length == 0) {
return;
}
int sum = 0;
this.sums[0] = 0;
for(int i = 0;i<len;i++){
sum += nums[i];
this.sums[i+1] = sum;
}
}
public int sumRange(int i, int j) {
return sums[j+1] - sums[i];
}
}