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

最新推荐文章于 2024-06-30 20:14:50 发布

原创最新推荐文章于 2024-06-30 20:14:50 发布 · 1.3k 阅读

2 ·

CC 4.0 BY-SA版权

动态规划专栏收录该内容

15 篇文章

订阅专栏

本文介绍了一种数组区间求和的高效算法实现。通过预处理数组，使用累积和的方法，可以将每次求和的时间复杂度从O(n)降低到O(1)，显著提升效率。

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

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

普通解法：

class NumArray {
private:
    vector<int> dp;
public:
    NumArray(vector<int>& nums) {
        dp = nums;
    }
    
    int sumRange(int i, int j) {
        int num = 0;
        while(i<=j)
        {
            num += dp[i];
            ++i;
        }
        return num;
    }
};

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

巧妙解法：

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

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