给定一个整数数组 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 方法。
思路:
这里题目肯定不希望看到我们从第i个元素累加到第j个元素这种做法,不然每次调用sumRange方法都会使用一个for循环,效率低下。
何不把求和一开始就算好,然后每次调用的时候做个减法就好?
/**
* @param {number[]} nums
*/
var NumArray = function(nums) {
this.array = [nums[0]];
for (let i = 1, len = nums.length; i < len; i++) {
this.array.push(nums[i] + this.array[i-1]);
}
};
/**
* @param {number} i
* @param {number} j
* @return {number}
*/
NumArray.prototype.sumRange = function(i, j) {
return (i === 0) ? this.array[j] : (this.array[j] - this.array[i-1]);
};
/**
* Your NumArray object will be instantiated and called as such:
* var obj = Object.create(NumArray).createNew(nums)
* var param_1 = obj.sumRange(i,j)
*/