题目
给定一个整数数组 nums,求出数组从索引 i 到 j (i ≤ j) 范围内元素的总和,包含 i, j 两点。
示例:
给定 nums = [-2, 0, 3, -5, 2, -1],求和函数为 sumRange()
sumRange(0, 2) -> 1
sumRange(2, 5) -> -1
sumRange(0, 5) -> -3
说明:
- 你可以假设数组不可变。
- 会多次调用 sumRange 方法。
代码
新鲜现做 幸福coding
class NumArray(object):
def __init__(self, nums):
"""
:type nums: List[int]
"""
self.dp = [0] * len(nums)
self.dp[0] = nums[0]
for i in range(1, len(nums)):
self.dp[i] = self.dp[i - 1] + nums[i]
def sumRange(self, i, j):
"""
:type i: int
:type j: int
:rtype: int
"""
if i == 0:
return self.dp[j]
else:
return self.dp[j] - self.dp[i-1]
# Your NumArray object will be instantiated and called as such:
nums = [-2, 0, 3, -5, 2, -1]
obj = NumArray(nums)
print nums
print obj.dp
print obj.sumRange(0, 2)
print obj.sumRange(2, 5)
print obj.sumRange(0, 5)
小结:动态规划。leetcode上测试case有问题,未能测试通过。仅供参考。