Leetcode 刷题 - 303 - Range Sum Query - Immutable

最新推荐文章于 2024-05-24 21:04:45 发布

加藤蜀黍

最新推荐文章于 2024-05-24 21:04:45 发布

阅读量182

点赞数

分类专栏： Python 文章标签： leetcode python

本文链接：https://blog.csdn.net/bearkino/article/details/50639747

版权

Python 专栏收录该内容

72 篇文章

订阅专栏

本文介绍了一种使用动态规划解决区间求和问题的方法，通过预处理得到累积和数组，进而快速计算任意指定区间的元素之和。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Given an integer array nums, find the sum of the elements between indices i and j (i ≤ j), inclusive.

Example:

Given nums = [-2, 0, 3, -5, 2, -1]

sumRange(0, 2) -> 1
sumRange(2, 5) -> -1
sumRange(0, 5) -> -3

Note:

You may assume that the array does not change.
There are many calls to sumRange function.

根据题意，就是需要求list的给定[start， end]的区间的sum值。

题目提示为DP，创建一个新的result list作为memory空间，记录到某个index位置的之前全部数字的sum值。

即，result[0, 1, 2, 3] = result[ nums[0], nums[0] + nums[1], nums[0] + nums[1] + nums[2], nums[0] + nums[1] + nums[2] + nums[3]]

所以，需要求得sumRange(i, j) 的时候，就是需要result[j] - result[i], 即：result[0, ..., j] - result[0, ..., i] = result[ i+1, ... ,j], 但是因为包含i, j。所以,需要加上index = i位置的值。最终就是result[j] - result[i] + nums[i]

class NumArray(object):
    def __init__(self, nums):
        """
        initialize your data structure here.
        :type nums: List[int]
        """
        self.nums = nums
        self.result = []
        sum = 0
        for i in range(len(nums)):
            sum += nums[i]
            self.result.append(sum)

    def sumRange(self, i, j):
        """
        sum of elements nums[i..j], inclusive.
        :type i: int
        :type j: int
        :rtype: int
        """
        return self.result[j] - self.result[i] + self.nums[i]