LeetCode刷题记录——第303题(区域和检索 - 数组不可变)

版权声明:此BLOG为个人BLOG,内容均来自原创及互连网转载。最终目的为记录自己需要的内容或自己的学习感悟,不涉及商业用途,转载请附上原博客。 https://blog.csdn.net/bulo1025/article/details/85677004

题目描述

给定一个整数数组 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 方法。

思路分析

  • 利用self.add记录累积和
  • 需要注意的是self.add的第0项为0,第一项为nums的第0项的和。所以要求第j项的和的时候需要add中第j+1项的值,这一点想清楚,然后减去add中第i项,因为算sumRange时nums中的第i项是要包括进来滴。

代码示例

class NumArray:

    def __init__(self, nums):
        """
        :type nums: List[int]
        """
        self.add = [0]
        for num in nums:
            self.add.append(self.add[-1] + num)
        

    def sumRange(self, i, j):
        """
        :type i: int
        :type j: int
        :rtype: int
        """
        return self.add[j+1] - self.add[i]
        
# Your NumArray object will be instantiated and called as such:
# obj = NumArray(nums)
# param_1 = obj.sumRange(i,j)
        

猜你喜欢

转载自blog.csdn.net/bulo1025/article/details/85677004