【LeetCode】第303题——区域和检索 - 数组不可变(难度:简单)
题目描述
给定一个整数数组 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))
提示:
0 <= nums.length <= 104
-105 <= nums[i] <= 105
0 <= i <= j < nums.length
最多调用 104 次 sumRange 方法
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/range-sum-query-immutable
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解题思路
该题预设了两个方法:用于初始化的NumArray()和用于返回数值和的sumRange()。
思路一:把循环放在sumRange()方法中,此方法适合多次初始化、少量返回和的场合,不适合本题中本人标黄处的描述。另外,此思路可能超时。
思路二:把循环放在NumArray()方法中,此方法适合少量初始化、多次返回和的场合,即满足题意。
思路一很容易想到,思路二是动态规划,使用HashMap作为存储与查找对象。不多废话,直接上代码。
代码详解
思路一:在sumRange()方法中写入循环(易超时)
class NumArray {
int[] list; // 用于接收nums
public NumArray(int[] nums) {
list = nums;
}
public int sumRange(int i, int j) {
long sum = 0; // 怕有中间值超过int范围而用long,但是其实用int修饰sum也行,因为得需保证任意ij范围内的和都得在int范围内
for(int index = i; index <= j; ++index) {
sum += list[index]; // 简单求和
}
return (int)sum;
}
}
/**
* Your NumArray object will be instantiated and called as such:
* NumArray obj = new NumArray(nums);
* int param_1 = obj.sumRange(i,j);
*/
思路二:在NumArray()方法中写入循环(推荐)
动态规划
class NumArray {
HashMap<Integer, Integer> hm = new HashMap<>(); // <下标i, 前i个(包含i)数值的和>
public NumArray(int[] nums) {
int sum = 0;
for(int i = 0; i < nums.length; ++i) {
sum += nums[i];
hm.put(i, sum);
// key=下标i,value=前i个(包含i)数值的和
// 如此一来,ij间的数值和就可以用j的value减去i-1的value即可
// 当然也得注意i=0这个特殊情况
}
}
public int sumRange(int i, int j) {
if(i == 0) { // 特殊情况
return hm.get(j); // 即前j个数(包含j)的和
} else {
return (hm.get(j) - hm.get(i-1)); // j的value减去i-1的value
}
}
}
注意点
- 注意求和包含下标 i 和 j,即包含边界。