Leetcode——303. 区域和检索 - 数组不可变

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

给定一个整数数组  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 方法。

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

解题思路:

1.确定状态

  • dp[i] 指的是前i个数之和为多少,sumRange 计算用dp[j+1] - dp[i]。

2.转移方程

  • dp[i]=dp[i-1]+nums[i-1] (i>2)

3.边界情况

  • 初始化为0
  • dp[1] = nums[0]

程序代码1:

class NumArray:
    def __init__(self, nums):
        len_nums = len(nums)
        if len_nums == 0:
            return
        self.dp = [0] * (len_nums+1)
        self.dp[1] = nums[0]
        for i in range(2,len_nums+1):
            self.dp[i] = self.dp[i-1] + nums[i-1]

    def sumRange(self, i, j):
        return self.dp[j+1] - self.dp[i]



nums = [-2, 0, 3, -5, 2, -1]
n = NumArray(nums)
print(n.sumRange(2,5))

程序代码2:

class NumArray:
    def __init__(self, nums):
        self.dp = [0]
        for num in nums:
            self.dp.append(self.dp[-1]+num)

    def sumRange(self, i, j):
        return self.dp[j+1] - self.dp[i]
发布了246 篇原创文章 · 获赞 155 · 访问量 14万+

猜你喜欢

转载自blog.csdn.net/suxiaorui/article/details/103570613