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

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

原始题目链接:https://leetcode.cn/problems/range-sum-query-immutable/

给定一个整数数组 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 <= 104
-105 <= nums[i] <= 105
0 <= i <= j < nums.length
最多调用 104 次 sumRange 方法

解题思路:

使用前缀和,前缀和适用场景是,原始数组不会被修改的情况下,频繁查询某个区间的累加和,它的大小是原数组大小+1,preSum[i] 代表nums[0…i-1] 所有元素的累加和,注意不包含i,计算索引 left 和 right (包含 left 和 right)之间的nums 元素的和,相当于计算 preSum[right+1] - preSum[left]。

计算前缀和的方式:
方法1:

n = len(nums)
presum = [0 for _ in range(n+1)]
for i in range(1, n+1):
    presum[i] = self.presum[i-1] + nums[i-1]

方法2:

sums = [0]
_sums = sums
for num in nums:
    _sums.append(_sums[-1] + num)

代码实现:

class NumArray:

    def __init__(self, nums: List[int]):
        if not nums:
            m = 0
        else:
            m = len(nums) + 1

        # 定义pre_sum[i]为:nums[0..i-1] 所有元素的累加和
        # 是不包含i的,前缀和的长度是原数组长度+1
        self.pre_sum = [0] * m
        for i in range(1, m):
            self.pre_sum[i] = self.pre_sum[i - 1] + nums[i - 1]


    def sumRange(self, left: int, right: int) -> int:
        return self.pre_sum[right + 1] - self.pre_sum[left]



# Your NumArray object will be instantiated and called as such:
# obj = NumArray(nums)
# param_1 = obj.sumRange(left,right)
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值