leetcode笔记之[303. 区域和检索]

42 阅读1分钟

携手创作,共同成长!这是我参与「掘金日新计划 · 8 月更文挑战」的第11天,点击查看活动详情

题目描述

原题链接 :

303. 区域和检索

给定一个整数数组  nums,处理以下类型的多个查询:

  1. 计算索引 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)