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

本文最后更新于:2022年4月9日 中午


给定一个整数数组 nums,求出数组从索引 ij (ij) 范围内元素的总和,包含 i, j 两点。

示例:

1
2
3
4
5
给定 nums = [-2, 0, 3, -5, 2, -1],求和函数为 sumRange()

sumRange(0, 2) -> 1
sumRange(2, 5) -> -1
sumRange(0, 5) -> -3

说明:

  1. 你可以假设数组不可变。
  2. 会多次调用 sumRange 方法。

Solution

  • 利用前缀和的思想
  • 建立新数组,将前缀和保存在新数组中

image-20201010135608994

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
# @lc code=start
class NumArray:
def __init__(self, nums: List[int]):
self.sums = [0]
for i in range(0, len(nums)):
self.sums.append(self.sums[i]+nums[i])

def sumRange(self, i: int, j: int) -> int:
return self.sums[j+1]-self.sums[i]


# Your NumArray object will be instantiated and called as such:
# obj = NumArray(nums)
# param_1 = obj.sumRange(i,j)
# @lc code=end

本博客所有文章除特别声明外,均采用 CC BY-SA 4.0 协议 ,转载请注明出处!