当前位置:首页 > 技术 > 正文内容

树状数组模板题解:区间操作与单点查询

访客 技术 2026年7月12日 2

树状数组(Binary Indexed Tree)是一种高效处理动态数组前缀和的数据结构,能够在O(log n)时间内完成单点修改和区间查询操作。根据应用场景的不同,树状数组主要分为两类基础模板:第一类支持单点修改与区间查询,第二类支持区间修改与单点查询。

模板一:单点修改 + 区间查询

该模板的核心思想是维护一个辅助数组,通过lowbit运算确定每个节点管理的区间范围。修改操作时,只需更新影响到的相关节点;查询前缀和时,将目标位置分解为多个管理区间的组合。

#include <bits/stdc++.h>
using namespace std;

const int MAXN = 500010;
int N, M;
int BIT[MAXN];

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

void update(int index, int delta) {
    while (index <= N) {
        BIT[index] += delta;
        index += getLowbit(index);
    }
}

int query(int index) {
    int sum = 0;
    while (index > 0) {
        sum += BIT[index];
        index -= getLowbit(index);
    }
    return sum;
}

int main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    
    cin >> N >> M;
    for (int i = 1; i <= N; ++i) {
        int val;
        cin >> val;
        update(i, val);
    }
    
    for (int i = 0; i < M; ++i) {
        int op;
        cin >> op;
        if (op == 1) {
            int pos, delta;
            cin >> pos >> delta;
            update(pos, delta);
        } else {
            int l, r;
            cin >> l >> r;
            cout << query(r) - query(l - 1) << '\n';
        }
    }
    return 0;
}

模板二:区间修改 + 单点查询

区间修改的实现利用了差分数组的思想。通过在区间起始位置加delta,区间结束位置加1处减delta,可以将区间修改转化为两个单点修改。查询时,将原数组值与差分数组的前缀和相加即可得到当前位置的实际值。

#include <bits/stdc++.h>
using namespace std;

const int MAXN = 500010;
int N, M;
int BIT[MAXN], arr[MAXN];

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

void update(int index, int delta) {
    while (index <= N) {
        BIT[index] += delta;
        index += getLowbit(index);
    }
}

int query(int index) {
    int sum = 0;
    while (index > 0) {
        sum += BIT[index];
        index -= getLowbit(index);
    }
    return sum;
}

int main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    
    cin >> N >> M;
    for (int i = 1; i <= N; ++i) {
        cin >> arr[i];
    }
    
    for (int i = 0; i < M; ++i) {
        int op;
        cin >> op;
        if (op == 1) {
            int l, r, delta;
            cin >> l >> r >> delta;
            update(l, delta);
            update(r + 1, -delta);
        } else {
            int pos;
            cin >> pos;
            cout << arr[pos] + query(pos) << '\n';
        }
    }
    return 0;
}

核心原理

树状数组的关键在于lowbit运算,它返回参数二进制表示中最低位的1所对应的值。每个树状数组节点管理从该位置向前lowbit长度个元素,通过这种分层结构,可以在logn时间内完成各种操作。

两种模板的区别仅在于应用场景的不同:第一种直接维护原数组的前缀和,第二种通过差分数组间接实现区间修改。理解差分数组与原数组的关系是掌握第二类模板的关键。

相关文章

Linux crontab 详解

1) crontab 是什么cron 是 Linux 的定时任务守护进程;crontab 是用来编辑/查看“按时间周期执行命令”的表(cron table)。常见两类:用户 crontab:每个用户一份(crontab -e 编辑)系统级 crontab / cron.d:可指定执行用户(/etc/crontab、/etc/cron.d/*)2) crontab 时间...

富文本里可以允许的 HTML 属性

一、所有标签默认允许的安全属性(极少)class        (可选)id           (通常建议禁用)title️ 注意:id 容易被滥用做锚点注入,很多系统直接禁用class 允许的话最好只允许固定前缀(如 editor-*)二、a 标签允许属性<a href="" t...

Dom\HTML_NO_DEFAULT_NS 的副作用:自动加闭合标签

在使用Dom\HTMLDocument时,Dom\HTML_NO_DEFAULT_NS 将禁止在解析过程中设置元素的命名空间, 此设置是为了与DOMDocument向后兼容而存在的。当使用它时,已知的一个副作用就是:自动加闭合标签例如 </img> 为什么会这样?当你使用:Dom\HTML_NO_DEFAULT_NS文档会变成 无命名空间模式,此时内部更接近 XML...

Laravel 事件和监听器创建

在 Laravel 中,使用 Artisan 命令创建 Events(事件) 和 Listeners(监听器) 是非常高效的。你可以通过以下几种方式来实现:1. 手动创建单个 Event如果你只想创建一个事件类,可以使用 make:event 命令:Bashphp artisan make:event UserRegistered执行后,文件将生成在 app/Even...

自定义域名解析神器 dnsmasq

什么是 dnsmasq?dnsmasq 是一个轻量级、功能强大的网络服务工具,专为小型和中等规模网络设计。它是一个综合的网络基础设施解决方案[1]。dnsmasq 能做什么?功能说明应用场景DNS 转发与缓存将 DNS 查询转发到上游服务器(ISP、Google DNS 等),并在本地缓存结果加快 DNS 查询速度,减少外部 DNS 流量本地 DNS解析本地网络设备的主机名,无需编辑&n...

linux screen 用法详情 (nohup 的替代方案)

一、screen 是什么?能干嘛?screen 是一个终端复用器,可以:在一个 SSH 会话中开多个“虚拟终端”SSH 断线后,程序仍然在后台运行随时重新连接到原来的会话特别适合:nohup 的替代方案跑脚本 / 爬虫 / 训练模型运维、远程开发二、安装 screen# CentOS / Rocky / Almayum install -y screen# Debian / Ubuntuapt i...

发表评论

访客

◎欢迎参与讨论,请在这里发表您的看法和观点。