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

题目描述

给定一个整数数组 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)