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

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

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/range-sum-query-immutable
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

 


 

  暴力解法,直接再__init__()中获取list,然后在成员函数中使用一个for循环进行累加,还好这道题不限制时间,代码如下:

  

 

 

   

 1 class NumArray:
 2 
 3     def __init__(self, nums: List[int]):
 4         self.nums = nums
 5         
 6     def sumRange(self, i: int, j: int) -> int:
 7         sum_ = 0
 8         for k in range(i, j+1):
 9             sum_ += self.nums[k]
10         return sum_

这样显然非常慢。

因此,可以有更有效的办法,就是初始化时,就补零,计算前i的累加和,再再成员函数中直接通过两个索引值相减便可得到结果:

通过补零的方法,可以使得两个list相同值的位置相互错开。--->>即前缀法

1 class NumArray:
2 
3     def __init__(self, nums: List[int]):
4         self.nums_sum = [0,]
5         for i in range(len(nums)):
6             self.nums_sum.append(self.nums_sum[i]+nums[i])
7 
8     def sumRange(self, i: int, j: int) -> int:
9         return self.nums_sum[j+1] - self.nums_sum[i]

 

 最后速度提升不少

posted @ 2021-03-01 09:09  ASTHNONT  阅读(63)  评论(0编辑  收藏  举报