Description
给定一个整数数组 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))
思路
标签里面给的是动态规划,但又是个easy,考虑使用半个动态规划的思想,即打表,用空间换时间(因为题目暗示要多次调用sumRange
,也就是为了避免TLE。
打表肯定不能打出所有子区间,那样肯定要炸空间。注意到sumRange
有两种情况:
s
u
m
R
a
n
g
e
[
0
,
j
]
=
∑
i
=
0
j
n
u
m
[
i
]
s
u
m
R
a
n
g
e
[
i
,
j
]
=
∑
i
=
i
j
n
u
m
[
i
]
sumRange[0,j]=\sum_{i=0}^j num[i]\\ sumRange[i,j]=\sum_{i=i}^j num[i]
sumRange[0,j]=i=0∑jnum[i]sumRange[i,j]=i=i∑jnum[i]
由于这是一个闭区间,即包括两端的数,所以考虑使用一个table,存储从开始到该数的和。这个list很好实现,遍历一次累加即可。对于sumRange
方法,考虑:
S
u
m
R
a
n
g
e
[
i
,
j
]
=
l
i
s
t
[
j
]
−
l
i
s
t
[
i
−
1
]
SumRange[i,j]=list[j]-list[i-1]
SumRange[i,j]=list[j]−list[i−1]
注意要包括两头。
代码
class NumArray:
def __init__(self, nums: List[int]):
if nums==[]:
return
self.nums=nums
self.sumindex=[self.nums[0]]
for i in range(1,len(self.nums)):# Sum is non-0 style
self.sumindex.append(self.sumindex[-1]+self.nums[i])
#print(self.sumindex)
def sumRange(self, i: int, j: int) -> int:
if i==0:
return self.sumindex[j]
else:
return self.sumindex[j]-self.sumindex[i-1]
# Your NumArray object will be instantiated and called as such:
# obj = NumArray(nums)
# param_1 = obj.sumRange(i,j)
执行用时:60 ms, 在所有 Python3 提交中击败了96.87%的用户
内存消耗:18.2 MB, 在所有 Python3 提交中击败了5.85%的用户
还行。。至于不打表怎么做到不TLE的还真的需要点本事。。。或许时间限制比较宽松?
打卡。
Upd:原来不需要存储nums
在self.nums
里面。。。