当前位置:

首页 > 编程开发 > 如何在 Java 中利用数组实现简单的二进制扇形树(Fenwick Tree)加速区间更新

如何在 Java 中利用数组实现简单的二进制扇形树(Fenwick Tree)加速区间更新

通过差分思想结合双树状数组维护d[i]和(i-1)*d[i]两个数组,实现区间加与区间求和操作,时间复杂度O(logn)。该算法将区间加转化为点加,区间求和通过前缀和差完成,适用于轻量级区间更新查询,代码简洁高效,空间复杂度低。

树状数组(Fenwick Tree)其实可以通过双数组配合差分思想,轻松实现区间更新与区间查询。核心思路是维护两个东西:d[i] 和 (i−1)·d[i]。时间复杂度稳定在 O(log n),非常适合那些只需要“区间加”和“区间求和”的轻量场景。

如何在 Ja va 中利用数组实现简单的二进制扇形树(Fenwick Tree)加速区间更新

其实 Ja va 里并没有“二进制扇形树”这个标准说法——你真正想用的,应该是树状数组(Fenwick Tree)。它基于整数的二进制低位(lowbit)来构造,主要用来高效处理前缀和、单点更新或区间更新查询。不过有一点要注意:经典的 Fenwick Tree 原生只支持单点更新 + 区间查询(或者反过来区间更新 + 单点查询),并不能直接同时支持“区间更新 + 区间查询”。如果想实现像 [l, r] 全部加 delta 并且还能求区间和,那就得引入差分思想,用两个树状数组来协同工作。

核心思路:差分 + 双树状数组

假设原数组是 a[1..n],先定义它的差分数组 d:

  • d[1] = a[1]
  • d[i] = a[i] − a[i−1] (i > 1)

这样一来,a[i] 就等于 d[1] + d[2] + … + d[i],也就是说,a 的前缀和其实就是 d 的前缀和。那么区间更新 a[l..r] += delta 就变成了:

 d[l] += delta,d[r+1] −= delta(如果 r+1 ≤ n)。

但问题来了:此时求 a[l..r] 的和,需要计算一个看起来很复杂的式子:

 ∑ₖ₌ₗʳ a[k] = ∑ₖ₌ₗʳ ∑ⱼ₌₁ᵏ d[j] = ∑ⱼ₌₁ʳ (r−j+1)·d[j] − ∑ⱼ₌₁ˡ⁻¹ (l−j)·d[j]

这事儿听起来有点绕——别担心。整理之后你会发现,它完全可以转化成两个前缀和的形式。于是我们就引入两个 Fenwick Tree:

  • tree1 负责维护 d[i]
  • tree2 负责维护 (i−1)·d[i]

说到底,只要维护好这两个数组,区间更新和区间查询就能同时搞定。

Ja va 实现:支持区间更新与区间查询的 Fenwick Tree

说到做到,Ja va 实现搞起来。以下是一个完整、可运行的数组实现(1-indexed,下标从 1 开始):

class FenwickRange {
    private long[] tree1;
    private long[] tree2;
    private int n;

    public FenwickRange(int size) {
        this.n = size;
        this.tree1 = new long[n + 1];
        this.tree2 = new long[n + 1];
    }

    // lowbit 运算:x & (-x)
    private int lowbit(int x) {
        return x & (-x);
    }

    // 向 tree 更新 index 位置 +delta
    private void update(long[] tree, int index, long delta) {
        while (index <= n) {
            tree[index] += delta;
            index += lowbit(index);
        }
    }

    // 查询 tree 前缀和 [1..index]
    private long query(long[] tree, int index) {
        long sum = 0;
        while (index > 0) {
            sum += tree[index];
            index -= lowbit(index);
        }
        return sum;
    }

    // 区间更新:a[l..r] += delta
    public void rangeUpdate(int l, int r, long delta) {
        // d[l] += delta → tree1[l] += delta, tree2[l] += (l-1)*delta
        update(tree1, l, delta);
        update(tree2, l, (long)(l - 1) * delta);
        // d[r+1] -= delta(若存在)
        if (r + 1 <= n) {
            update(tree1, r + 1, -delta);
            update(tree2, r + 1, (long) r * (-delta));
        }
    }

    // 查询 a[1..x] 的前缀和
    private long prefixSum(int x) {
        return query(tree1, x) * x - query(tree2, x);
    }

    // 查询 a[l..r] 的区间和
    public long rangeSum(int l, int r) {
        return prefixSum(r) - prefixSum(l - 1);
    }
}

使用示例也非常直观:

public class Main {
    public static void main(String[] args) {
        FenwickRange ft = new FenwickRange(5);
        // 初始数组 a = [0,1,2,3,4,5](1-indexed,a[1]=1…a[5]=5)
        // 等价于初始化差分:d[1]=1, d[2]=1, d[3]=1, d[4]=1, d[5]=1
        ft.rangeUpdate(1, 1, 1); // a[1] += 1 → a[1]=2
        ft.rangeUpdate(2, 4, 10); // a[2..4] += 10 → a=[2,12,13,14,5]
        System.out.println(ft.rangeSum(2, 4)); // 输出 12+13+14 = 39
    }
}

关键细节与注意事项

  • 下标必须 1-indexed:Fenwick Tree 太依赖 lowbit 了,如果用下标 0 会导致死循环。所以数组长度声明为 n+1,操作时索引从 1 开始,这一点踩坑的人不少。
  • 数据类型用 long:避免区间更新时中间乘法溢出(比如 (l−1)×delta),用 long 更安全。
  • rangeUpdate 中 r+1 越界检查不能省:不然一不小心就写到了 tree[n+1] 引发 ArrayIndexOutOfBoundsException。
  • 单点更新/查询可以复用:比如单点更新 a[i] += v 就是 rangeUpdate(i,i,v);单点查询 a[i] 就是 rangeSum(i,i)。
  • 所有操作的时间复杂度都是 O(log n),空间 O(n),用起来很轻快。

对比线段树:什么时候选 Fenwick?

和线段树比起来,Fenwick Range 的实现明显更轻量,代码更短,常数也更小。它特别适合下面这些场景:

  • 只需要区间加、区间求和,不涉及赋值、取 max/min、历史版本等复杂操作。
  • 离线操作或者强制在线但内存敏感的环境。
  • 竞赛场景中追求简洁与速度,Fenwick 往往是首选。

当然,如果你的需求更复杂,比如需要区间赋值、多种标记下传、动态开点,那线段树仍然是更通用的选择。Fenwick 和线段树各有千秋,选哪个,关键看你的数据结构和操作到底有多“单纯”。

本站声明:本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系bd@zhengruan.com
作者最新文章
编程开发 Java
相关文章 更多
Java抽象方法修饰符限制:为何不能用private或static
Java抽象方法修饰符限制:为何不能用private或static

深入剖析Java中抽象方法为何严禁使用private和static修饰符。从面向对象契约规范、方法继承重写、JVM动态绑定(invokevirtual)与静态解析(invokestatic)等底层机制展开详细推导,结合编译报错示例与全景对比,彻底理清修饰符限制背后的设计逻辑。

Java变量声明与初始化:局部、成员及静态变量的定义语法
Java变量声明与初始化:局部、成员及静态变量的定义语法

系统解析Java局部变量、成员变量与静态变量的定义语法、声明与初始化机制、生命周期及JVM内存驻留规则,结合图解与代码示例分析常见编译错误与认知误区。

Java子类未实现抽象方法编译错误修复指南
Java子类未实现抽象方法编译错误修复指南

针对Java开发中常见的“子类未实现抽象方法”编译错误,深入分析报错原因,提供重写实现、声明抽象子类两种标准修复路径,并总结参数签名、访问修饰符等典型避坑要点。

junit怎么用零基础完成第一个单元测试教程
junit怎么用零基础完成第一个单元测试教程

适合初学者的JUnit入门指南,详解如何配置环境、编写首个测试用例、使用断言验证结果,并分析单元测试的适用场景与例外情况。

Java发送post请求发送json数据的实现方法与代码示例
Java发送post请求发送json数据的实现方法与代码示例

深入解析Java发送JSON格式POST请求的实现机制,对比原生API与第三方库的代码差异,提供可复制的代码示例,并探讨超时设置、编码处理及微服务场景下的例外情况。

百度官网入口最新版 百度搜索网页版直达
百度官网入口最新版 百度搜索网页版直达

百度官方在线入口最新版网页版直达官网是https://www.baidu.com/,界面简洁无干扰、多端同步体验一致、内容聚合维度丰富、语音与图像识别稳定、隐私控制机制可见。

谷歌浏览器Mac版入口
谷歌浏览器Mac版入口

谷歌浏览器Mac版官方安装指南 谷歌浏览器Mac版官方安装入口是https://www.google.com/chrome/,需macOS 12+系统、500MB空间,下载.dmg后拖入应用程序安装,支持多设备同步、性能优化与隐私保护功能。 苹果电脑Chrome的安装入口究竟在哪里?这个问题最近可是

Chrome浏览器JS脚本不运行怎么办
Chrome浏览器JS脚本不运行怎么办

Chrome中JavaScript未执行需依次检查:一、移除站点级禁用并添加允许域名;二、开启全局JavaScript开关;三、禁用干扰扩展;四、在开发者工具中启用JavaScript;五、重置内容设置为默认。 有时在Chrome里打开网页,会发现交互按钮点了没反应,数据加载不出来,页面仿佛“静止”

IE浏览器怀旧版在线网址
IE浏览器怀旧版在线网址

IE浏览器怀旧版在线网址:一次精准的技术时光回溯 最近,不少老用户和怀旧爱好者在反复搜索一个问题:那个经典的Internet Explorer,如今还能在哪里原汁原味地体验到?答案指向一个特定的地址:https://ie.microsoft.com/legacy/。 这个网站远不止是一个简单的“皮肤

火狐浏览器有哪些设置功能
火狐浏览器有哪些设置功能

火狐浏览器五大核心设置功能:解锁高效、安全与个性化体验 火狐浏览器功能强大,但如果不仔细挖掘,很多能大幅提升效率和安全性的设置可能就“藏着掖着”了。这就好比拥有一台高性能设备,却只用了基础模式。那么,如何把它调整到最顺手、最安全的状态?接下来,我们就聚焦于当前版本(截至2025年末)最关键的五大设置

查看更多
精品专题 更多
装机必备
装机必备

正软商城装机必备专区,精选办公、浏览器、安全防护、影音播放、压缩解压、设计创作和系统工具等电脑常用正版软件,帮助用户快速完成新电脑软件配置。

Windows
Windows

正软商城Windows软件专区,汇集适用于Windows电脑的办公、设计、安全防护、影音播放、开发工具和系统优化软件,提供软件介绍、系统要求、正版授权及购买下载服务。

PDF教程
PDF教程

正软商城PDF教程频道提供PDF编辑、转换、合并、拆分、压缩及格式处理方法,同时介绍常用PDF软件和工具的使用技巧。

Mac软件 更多
Postman macOS版
Postman macOS版
Mac

Postman是一套面向API全生命周期的开发与测试平台,可在Mac上发送和调试多种协议请求,管理集合、环境及认证信息,并完成脚本测试、模拟服务、监控、文档发布和团队协作。它适合开发、测试及运维人员使用,提供个人免费方案。

Docker Desktop macOS版
Docker Desktop macOS版
Mac

Docker Desktop 是 Docker 官方为 Mac 提供的容器开发环境,集成 Docker Engine、命令行工具、Compose、Build、Kubernetes 与图形化管理界面。

Android Studio macOS版
Android Studio macOS版
Mac

Android Studio 是 Google 面向安卓应用开发提供的官方集成开发环境,基于 IntelliJ IDEA 构建。它整合智能代码编辑、Gradle 构建、界面预览、设备模拟、性能分析、调试测试及 APK 分析等工具。

WINDOWS 更多
PixPix 网页版
PixPix 网页版
Windows

PixPix 是面向品牌、卖家和内容团队的在线 AI 图片视频创作平台,提供图视频 Agent、无限画布、图片生成、视频生成、自然语言编辑及商品套图、产品精修和带货视频等电商工具。

Postman Windows版
Postman Windows版
Windows

Postman 是面向 API 全生命周期的开发与协作平台。Windows 桌面版可用于构建和发送 HTTP、GraphQL、gRPC、WebSocket、MQTT 等请求,并提供环境变量、集合、脚本测试、模拟服务、监控、文档及团队工作区等

Docker Desktop Windows版
Docker Desktop Windows版
Windows

Docker Desktop 为 Windows 开发者提供集成式容器环境,内置 Docker Engine、命令行工具、Compose、Build 与 Kubernetes 等组件,可在本机完成镜像构建、容器运行、日志查看和资源管理。