题目

给定一个整数数组 nums，求出数组从索引 i 到 j（i ≤ j）范围内元素的总和，包含 i、j 两点。

实现 NumArray 类：

  • NumArray(int[] nums) 使用数组 nums 初始化对象
  • int sumRange(int i, int j) 返回数组 nums 从索引 i 到 j（i ≤ j）范围内元素的总和，包含 i、j 两点（也就是 sum(nums[i], nums[i + 1], … , nums[j])）

示例：

输入：
[“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))

提示：

  • 0 <= nums.length <= 10e4
  • -10e5 <= nums[i] <= 10e5
  • 0 <= i <= j < nums.length
  • 最多调用 10e4 次 sumRange 方法

题解

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
class NumArray:

    def __init__(self, nums: List[int]):
        if not nums:
            return 
        self.numsa = nums
        self.pre_nums = [nums[0]]
        for i in range(1,len(nums)):
            self.pre_nums.append(self.pre_nums[-1]+nums[i])
        

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


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