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)