303. 区域和检索 - 数组不可变

题目

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

前缀和

class NumArray:
    def __init__(self, nums: List[int]):
        s = [0] * (len(nums) + 1) #初始化前缀和数组,长度比原始数组多1
        for i, x in enumerate(nums): #i表示下标,x表示下标对应的值
            s[i + 1] = s[i] + x  # 计算前缀和,累加前一个前缀和和当前值
        self.s = s  # 将前缀和数组存储在实例变量中
    def sumRange(self, left: int, right: int) -> int:
        return self.s[right + 1] - self.s[left] #返回指定区间的和,通过前缀和数组计算
posted @ 2024-03-19 09:55  Frommoon  阅读(13)  评论(0)    收藏  举报