给定一个整数数组 nums,求出数组从索引 i 到 j(i ≤ j)范围内元素的总和,包含 i, j 两点。
实现NumArray类:
NumArray(int[] nums)
用整数数组nums初始化对象。int sumRange(int i, int j)
返回nums数组的元素之和,范围为[i,j](包括sum(nums [i],nums [i + 1],…,nums [j]))
提示:
- 0 <= nums.length <= 104
- -105 <= nums[i] <= 105
- 0 <= i <= j < nums.length
- 最多调用 104 次 sumRange 方法
解法一
暴力
class NumArray {
private int[] data;
public NumArray(int[] nums) {
data = nums;
}
public int sumRange(int i, int j) {
int sum = 0;
for (int n = i; n <= j; n++) {
sum += data[n];
}
return sum;
}
}
/**
* Your NumArray object will be instantiated and called as such:
* NumArray obj = new NumArray(nums);
* int param_1 = obj.sumRange(i,j);
*/
解法二
前缀和
class NumArray {
private int[] data;
public NumArray(int[] nums) {
data = new int[nums.length + 1];
data[0] = 0;
for (int i = 0; i < nums.length; i++) {
//data数组的i+1位,保存nums数组前i个元素之和
data[i + 1] = data[i] + nums[i];
}
}
public int sumRange(int i, int j) {
//如有1,2,3,4,5
//i为2,j为5
//nums的前5(因为data的i+1保存nums的前i元素之和,所以这里要加1)元素之和减去nums的前1(同理,因为2是需要的,而data[2]为nums前1个之和)元素之和,等于2到5的和
return data[j + 1] - data[i];
}
}
看了官方的解法三,用的是前缀和,又get到新思路。