数据结构 -- 树状数组

前言

树状数组或二叉索引树(Binary Indexed Tree),又以其发明者命名为 Fenwick 树。其初衷是解决数据压缩里的累积频率的计算问题,现多用于高效计算数列的前缀和、区间和。它可以以 O(logn) 的时间得到任意前缀和。并同时支持在 O(logn) 时间内支持动态单点值的修改。空间复杂度 O(n)。

介绍

1.原理

树状数组,顾名思义是树状的数组,我们首先引入二叉树,叶子节点代表A1~A8

现在变形一下:

现在定义每一列的顶端节点C数组(其实C数组就是树状数组),如图:

理解树状数组的重点

Ci代表子树的叶子节点的权值之和,如图可以知道:

C1=A1;

C2=A1+A2;

C3=A3;

C4=A1+A2+A3+A4;

C5=A5;

C6=A5+A6;

C7=A7;

C8=A1+A2+A3+A4+A5+A6+A7+A8;

看过上面的列表,有没有看出来什么规律来呢?

如果没有看出来,那说明你不知道lowbit函数。

先看lowbit函数的代码:

javascript 复制代码
function lowBit(x){
    return x&-x;
}

不知道干啥的吧!是不是一脸懵逼?别慌,举几个例子:

1 & -1 = 1;

2 & -2 = 2;

3 & -3 = 1;

4 & -4 = 4;

5 & -5 = 1;

6 & -6 = 2;

7 & -7 = 1;

8 & -8 = 8;

对比上面的,熟悉不,看出来规律了吗?

怎么理解呢?

lowbit函数的含义即为:x(二进制)最低位1后面的0组成的数

例如:

5(二进制:101)最低位1后面0的个数为0,则lowbit(5)=2^0=1(十进制)

12(二进制:1100) 最低位1后面0的个数为2,则lowbit(12)=2^2=4(十进制)

那现在我给你一个数组,让你求解他的树状数组,能求出来吗?思考一下解出来了就跳过下面。或者对一下答案!

javascript 复制代码
var A = [1,2,3,4,5,6,7,8]

var C = new Array(9).fill(0);

function getSum(i) {  
    let begin = lowBit(i);
	let res=0;
	while(begin) {
		res += A[i-begin];
		begin --;
	}
	return res;
}

for(let i=1;i<9;i++){
    C[i] = getSum(i);
}
console.log("树状数组",treeArr);
//[0,1, 3, 3, 10, 5, 11, 7, 36]
//注意哦我不要第一项,至于为什么,懂得都懂。

2.应用

2.1 区间查询

利用Ci数组,求A数组中前i项和,举两个栗子:

前7项和:

sum7=A1+A2+A3+A4+A5+A6+A7

而C4=A1+A2+A3+A4;C6=A5+A6;C7=A7

可以得到:sum7=C4+C6+C7

数组下标写成二进制:sum(111)=C(100)+C(110)+C(111)
前5项和:

sum5=A1+A2+A3+A4+A5

而C4=A1+A2+A3+A4;C5=A5

可以得到:sum5=C4+C5

数组下标写成二进制:sum(101)=C(100)+C(101)

细细观察二进制,树状数组追其根本就是二进制的应用,结合代码演示一下代码过程:

javascript 复制代码
function finSum(i){
    let res = 0;
    while(i){
        res += treeArr[i];
        i -= lowBit(i)
    }
    return res;
}
console.log("前5项和",finSum(5)); //15
console.log("前7项和",finSum(7)); //28

代码推演:

lowbit(5)=001 5-lowbit(5)=4(100) ans+=C4

lowbit(4)=100 4-lowbit(4)=0(000) break;

看到这里是不是有点疑惑,这不比前缀算法还多算了一个树状数组吗?我也有这个疑问,保留意见,先继续往下看。

2.2 单点更新

当我们修改A数组中某个值时,应当如何更新C数组呢?回想一下,区间查询的过程,再看一下上文中列出的过程。这里声明一下:单点更新实际上是不修改A数组的,而是修改树状数组C,向上更新区间长度为lowbit(i)所代表的节点的值。

javascript 复制代码
function update(i,k){
    A[i] += k;
    while(i <= 8){
       	C[i] += k;
        i += lowBit(i);
    }
}
update(6,5);

console.log("前5项和",finSum(5)); //15
console.log("前7项和",finSum(7)); //33

如上图:若在A1加上值val,即更新A1时,需要向上更新C1,C2,C4,C8,这个时候只需将这4个节点每个节点的值加上val即可。这里为了方便大家理解,人为添加了个A数组表示每个叶子节点的值,事实上A数组并不用修改,实际运用中也可不设置A数组,单点更新只需修改树状数组C即可。下标写成二进制:C(001),C(010),C(100),C(1000)

lowbit(1)=001 1+lowbit(1)=2(010) C2+=val;

lowbit(2)=010 2+lowbit(2)=4(100) C4+=val;

lowbit(4)=100 4+lowbit(4)=8(1000) C8+=val;

由于c1 c2 c4 c8 都包含有A1,所以在更新A1时实际上就是更新每一个包含A1的节点。

现在是不是有点懵懂了,如果用前缀和的话,需要在1后面的所有元素都加上val,但是对于树状数组就不同了。时间复杂度为O(logN)

与线段树对比

1.两者在复杂度上同级, 但是树状数组的常数明显优于线段树, 其编程复杂度也远小于线段树.

2.树状数组的作用被线段树完全涵盖, 凡是可以使用树状数组解决的问题, 使用线段树一定可以解决, 但是线段树能够解决的问题树状数组未必能够解决.

  1. 树状数组的突出特点是其编程的极端简洁性, 使用lowbit技术可以在很短的几步操作中完成树状数组的核心操作,其代码效率远高于线段树。
相关推荐
cpp_25011 小时前
P10722 [GESP202406 六级] 二叉树
数据结构·c++·算法·题解·洛谷·树形结构·gesp六级
smj2302_796826521 小时前
解决leetcode第3948题字典序最大的MEX数组
python·算法·leetcode
周末也要写八哥2 小时前
浅谈:C++中cpp 14 ~ cpp 17
开发语言·c++·算法
许彰午2 小时前
13_HashMap底层原理详解
算法·哈希算法
GIOTTO情2 小时前
基于 NLP 情感加权算法的智能舆情处置系统架构与落地实现
人工智能·算法·自然语言处理
无风听海2 小时前
PKCE 的 S256 算法深度剖析:从协议设计到密码学原理
javascript·网络·算法·密码学
凌波粒2 小时前
LeetCode--530.二叉搜索树的最小绝对差(二叉树)
算法·leetcode·职场和发展
sali-tec2 小时前
C# 基于OpenCv的视觉工作流-章82-毛刺检测
图像处理·人工智能·opencv·算法·计算机视觉
恒想进步2 小时前
leetcode202.快乐数
算法
8Qi83 小时前
LeetCode 208:实现 Trie(前缀树)—— Java 题解 ✅
java·算法·leetcode·二叉树·tire树