力扣每日一题-区域和检索 - 数组可修改

307. 区域和检索 - 数组可修改 - 力扣(LeetCode)

看题面单点修改加区间查询 == 一眼线段树 >> 由于没有复杂修改操作 - > 简化为树状数组求解

cpp 复制代码
int init = []()
{
    cin.tie(0) -> sync_with_stdio(false);
    return 0;
}();

class NumArray {
public:
    NumArray(vector<int>& nums) : sum(nums.size() + 1, 0), num(nums), n(nums.size()) 
    {
        for(int i = 0;i < n;i++)
        add(i + 1,nums[i]);
    }

    int lowbit(int x)
    {
        return x & -x;
    }

    void add(int pos, int val)
    {
        for(; pos <= n; pos += lowbit(pos))
        sum[pos] += val;
    }

    void update(int index, int val) 
    {
        add(index + 1, val - num[index]);
        num[index] = val;
    }
    
    int ask(int x)
    {
        int ans = 0;
        for(; x > 0; x -= lowbit(x))
        ans += sum[x];
        return ans;
    }

    int sumRange(int left, int right) 
    {
        return ask(right + 1) - ask(left);
    }
private:
vector<int>sum;
vector<int>&num;
int n;
};

/**
 * Your NumArray object will be instantiated and called as such:
 * NumArray* obj = new NumArray(nums);
 * obj->update(index,val);
 * int param_2 = obj->sumRange(left,right);
 */

还有一个有意思的解法

cpp 复制代码
int init = []()
{
    cin.tie(0) -> sync_with_stdio(false);
    return 0;
}();

class NumArray 
{
public:
    NumArray(vector<int>& nums) 
    {
        n = nums.size();
        v.resize(n << 1);
        for(int i = n; i < (n << 1); i++)
        v[i] = nums[i - n];
        for(int i = n - 1; i > 0; i--)
        v[i] = v[i << 1] + v[(i << 1) + 1];
    }
    
    void update(int index, int val) 
    {
        int i = n + index;
        int diff = val - v[i];
        v[i] = val;
        while(i > 1)
        {
            v[i >> 1] += diff;
            i >>= 1;
        }
    }
    
    int sumRange(int left, int right) 
    {
        left += n;right += n; 
        int sum = 0;
        while(left <= right)
        {
            if(left & 1){ sum += v[left]; left++; }
            if(!(right & 1)){ sum += v[right]; right--; }  
            left >>= 1;right >>= 1;
        }
        return sum;
    }
private:
    vector<int> v;
    int n;
};
相关推荐
SweetCode8 分钟前
裴蜀定理:整数解的奥秘
数据结构·python·线性代数·算法·机器学习
ゞ 正在缓冲99%…21 分钟前
leetcode76.最小覆盖子串
java·算法·leetcode·字符串·双指针·滑动窗口
xuanjiong22 分钟前
纯个人整理,蓝桥杯使用的算法模板day2(0-1背包问题),手打个人理解注释,超全面,且均已验证成功(附带详细手写“模拟流程图”,全网首个
算法·蓝桥杯·动态规划
惊鸿.Jh41 分钟前
【滑动窗口】3254. 长度为 K 的子数组的能量值 I
数据结构·算法·leetcode
明灯L42 分钟前
《函数基础与内存机制深度剖析:从 return 语句到各类经典编程题详解》
经验分享·python·算法·链表·经典例题
碳基学AI1 小时前
哈尔滨工业大学DeepSeek公开课:探索大模型原理、技术与应用从GPT到DeepSeek|附视频与讲义免费下载方法
大数据·人工智能·python·gpt·算法·语言模型·集成学习
补三补四1 小时前
机器学习-聚类分析算法
人工智能·深度学习·算法·机器学习
独好紫罗兰1 小时前
洛谷题单3-P5718 【深基4.例2】找最小值-python-流程图重构
开发语言·python·算法
正脉科工 CAE仿真1 小时前
基于ANSYS 概率设计和APDL编程的结构可靠性设计分析
人工智能·python·算法
爱喝热水的呀哈喽1 小时前
Java 集合 Map Stream流
数据结构