- 题目描述
给定一个整数数组 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 方法。
- 解决思路一
按我自己肤浅的想法呢,真的是没想到这个题和动态规划的关系到底是个啥,所以想到的是就是八竿子打不着的解决方案,不过也是学到了,可以用sum()函数来对List求和,把List做它的参数即可。
- 代码一
class NumArray(object):
def __init__(self, nums):
"""
:type nums: List[int]
"""
self.nums = nums
def sumRange(self, i, j):
"""
:type i: int
:type j: int
:rtype: int
"""
return sum(self.nums[i:j+1])
- 解决思路二
其实在区域求和的部分,是可以用到动态规划的思想的,如果我们要求数组中第i个元素到第j个元素的和,那么如果我们一直维护着一个保存从第1个元素到目前元素的和的数组dp[n],那么直接返回dp[j] - dp[i-1]即可
- 代码二
class NumArray(object):
def __init__(self, nums):
"""
:type nums: List[int]
"""
self.dp = [0] *len(nums)
sum = 0
for i in range(len(nums)):
sum += nums[i]
self.dp[i] = sum
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]
- ps:虽然我没用,但是参考的代码里用到了python类、protected成员、私有成员的概念,还是很有用的,记一下。
_var ;变量名前一个下划线来定义,此变量为保护成员protected,只有类及其子类可以访问。此变量不能通过from XXX import xxx 导入
__var;变量名前两个下划线来定义,此变量为私有private,只允许类本身访问,连子类都不可以访问。
class NumArray:
def __init__(self, nums):
"""
:type nums: List[int]
"""
self.__dp = [0] * len(nums)
sums = 0
for i in range(len(nums)):
sums += nums[i]
self.__dp[i] = sums
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]