给定一个整数数组 nums
,求出数组从索引 i
到 j(i ≤ j)
范围内元素的总和,包含 i
、j
两点。
实现 NumArray 类:
NumArray(int[] nums) 使用数组 nums 初始化对象
int sumRange(int i, int j) 返回数组 nums 从索引 i 到 j(i ≤ j)范围内元素的总和,包含 i、j 两点(也就是
sum(nums[i], nums[i + 1], ... , nums[j]))
示例:
输入:
["NumArray", "sumRange", "sumRange", "sumRange"]
[[[-2, 0, 3, -5, 2, -1]], [0, 2], [2, 5], [0, 5]]
输出:
[null, 1, -1, -3]
解释:
NumArray numArray = new NumArray([-2, 0, 3, -5, 2, -1]);
numArray.sumRange(0, 2); // return 1 ((-2) + 0 + 3)
numArray.sumRange(2, 5); // return -1 (3 + (-5) + 2 + (-1))
numArray.sumRange(0, 5); // return -3 ((-2) + 0 + 3 + (-5) + 2 + (-1))
简单题考虑直接相加:
class NumArray {
int[] nums;
public NumArray(int[] nums) {
if(nums.length == 0) return;
this.nums = nums;
}
public int sumRange(int i, int j) {
int sum = 0;
for (int k = i; k <= j; k++) {
sum += nums[k];
}
return sum;
}
}
/**
* Your NumArray object will be instantiated and called as such:
* NumArray obj = new NumArray(nums);
* int param_1 = obj.sumRange(i,j);
*/
一看效率低的不行我就知道不简单,原来还有一种思路,在直接在构造函数维护一个数字,每个数组的值是前面所有preSum,这样计算两个位置和就变成数组位置相减 秒啊 知道原理 这样的简单代码还是可以的。
class NumArray {
int[] nums;
int[] preSum;
public NumArray(int[] nums) {
if(nums.length == 0) return;
this.nums = nums;
preSum = new int[nums.length + 1];
for (int i = 0; i < nums.length; i++) {
preSum[i+1] = preSum[i] + nums[i];
}
}
public int sumRange(int i, int j) {
return preSum[j+1] - preSum[i];
}
}
搞定!