LeetCode探索之旅（79）-303Range sum

最新推荐文章于 2024-09-03 22:33:53 发布

原创最新推荐文章于 2024-09-03 22:33:53 发布 · 208 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#LeetCode #动态规划

代码训练专栏收录该内容

177 篇文章

订阅专栏

本文深入解析LeetCode第303题的解决方案，介绍了一种利用动态规划思想预先计算前缀和，进而高效求解任意区间元素之和的方法。通过C++与Python代码实现，展示了如何在O(1)时间内完成区间和的计算。

今天继续刷LeetCode，第303题，求数组中一个区间的数和。

分析：
类似于动态规划，先计算前面累加的和，然后通过相减求区间和。

问题：
1、动态规划；

附上C++代码：

class NumArray {
public:
    NumArray(vector<int>& nums) {
        if(nums.empty())
            return;
        else
        {
            sum.push_back(nums[0]);
            int len=nums.size();
            for(int i=1;i<len;i++)
            {
                sum.push_back(nums[i]+sum[i-1]);
            }
        }
    }
    
    int sumRange(int i, int j) {
        if(0==i)
            return sum[j];
        int len=sum.size();
        if(i<0||i>=len||j<0||j>=len||i>j)
            return 0;
        return sum[j]-sum[i-1];
    }
    
private:
    vector<int> sum;
};

/**
 * Your NumArray object will be instantiated and called as such:
 * NumArray* obj = new NumArray(nums);
 * int param_1 = obj->sumRange(i,j);
 */

附上 Python代码：

class NumArray:

    def __init__(self, nums: List[int]):
        self.dp=nums
        for i in range(1,len(nums)):
            self.dp[i]+=self.dp[i-1]

    def sumRange(self, i: int, j: int) -> 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:
# obj = NumArray(nums)
# param_1 = obj.sumRange(i,j)