携手创作,共同成长!这是我参与「掘金日新计划 · 8 月更文挑战」的第11天,点击查看活动详情
题目描述
原题链接 :
给定一个整数数组 nums,处理以下类型的多个查询:
- 计算索引 left 和 right (包含 left 和 right)之间的 nums 元素的 和 ,其中 left <= right
实现 NumArray 类:
- NumArray(int[] nums) 使用数组 nums 初始化对象
- int sumRange(int i, int j) 返回数组 nums 中索引 left 和 right 之间的元素的 总和 ,包含 left 和 right 两点(也就是 nums[left] + nums[left + 1] + ... + nums[right] )
示例 1:
输入:
["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))
提示:
- 1 <= nums.length <= 10^4
- -10^5 <= nums[i] <= 10^5
- 0 <= i <= j < nums.length
- 最多调用 10^4 次 sumRange 方法
思路分析
基本思路:根据提示多次调用,想到将前k项之和(0<=k<=max)记录在表中,当调用计算函数时,只需找到对应记录位置相减即可
说一下两个易错点:
一开始我将动态规划的代码写到计算函数中去了,这是错误的,因为每次调用该函数都会重复计算,导致超时
sums表中初始化一开始设置的是nums[0],这样会导致一种情况即nums为空时出错,所以初始化为0,后续只要理解思路就可以设置相应的变量范围
AC 代码
class NumArray:
# 思路:多次调用,如果暴力解答,可能超时,所以可以事先将前n项之和计算出来,相减就是所求结果
def __init__(self, nums: List[int]):
self.nums = nums
self.sums = [0]
for t in range(0, len(self.nums)):
# 当前项加上前一步计算出来的和,该处体现动态规划思想
self.sums.append(self.nums[t] + self.sums[t])
# 不能把动态规划的东西写在这个函数中,因为多次调用会导致多次计算前n项之和
def sumRange(self, i: int, j: int) -> int:
return self.sums[j+1] - self.sums[i]
# Your NumArray object will be instantiated and called as such:
# obj = NumArray(nums)
# param_1 = obj.sumRange(i,j)