树状数组概览

目录

目录

附录

Part 1:What is 树状数组？

Part 2:树状数组适合解决什么样的问题？

Part 3:如果我是初学者，树状数组中的那些基本代码片段需要我学习？

1.树状数组的工作原理和lowbit函数

I.推导

II.解决 

 2.线段树的创建和modify函数

3.线段树的查询和query函数

Part 4:总结

附录

聊完了线段树，我们来看看树状数组

想要了解线段树的朋友，可以看看我的另一篇博客：线段树概览-CSDN博客

Part 1:What is 树状数组？

和线段树类似，也利了用二叉思想。区别在于线段树是一种二叉搜索树，而树状数组是一种二叉索引树，它用一个数组模拟了树结构。这种数据结构主要用于快速查找、‌插入和删除操作。‌

如果没看懂，我们再来看图：

这是一个标准的树状数组（这个图是我手绘的你怎么又不懒了，如果要取用的话请告知我），可以看出，虽然它采用了二叉树的思想，但它不完全是二叉树，具体将由一个tree数组实现，会在后面讲到。

p.s.实际上你遇到的树状数组可能是这样的：

然而这并不影响我们做题。

Part 2:树状数组适合解决什么样的问题？

树状数组的适用题目和线段树大致一致，都是区间查询类问题，但比线段树更适合一些简单题（代码量少，实现简单）以及空间复杂度要求较高的题。

Part 3:如果我是初学者，树状数组中的那些基本代码片段需要我学习？

（此处的部分代码片段的取名为个人喜好，其他题解里或许不一样，无需模仿）

1.树状数组的工作原理和lowbit函数

I.推导

树状数组的工作原理相当于一个线性的更新，仔细观察我们的图，这里再放一遍：

还是看不出来什么呀我们分析每个子节点及其祖先节点的二进制：

[第一组]                  [第二组]

1:00000001            5:00000101

2:00000010            6:00000110

4:00000100            8:00001000

8:00001000

我们发现， 每一组的子节点及其最近公共祖先的下标关系是：

其最近公共祖先下标=其下标+其下标在二进制下最后一个1所代表的数

所以问题直接转变为了如何求解一个数在二进制下最后一个1所代表的数。

II.解决 

想要解决这个问题需要引入补码知识。我们知道，一个数的二进制表示可以被我们称为源码，其反码（即这个数的相反数的二进制）为源码每一位上的二进制数与1异或得到的结果（即1变成0，0变成1），补码就是反码+1得到的值。下面是两组例子：

[第一组]                                             [第二组]

源码:00000011 （3的二进制）          源码:00001000 （8的二进制）

反码:11111100  （-3的二进制）        反码:11110111 （-8的二进制）

补码:11111101                                   补码:11111000

可以发现，源码与补码按位与（&）得到的结果就是我们想要得到的值。由此写出以下代码：

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

 2.线段树的创建和modify函数

这一部分需要用到lowbit函数，如果还没学会或已经忘了请回到上面重新食用。

这一部分在上面已经推导过了，这里再把结论放一下：

其最近公共祖先下标=其下标+其下标在二进制下最后一个1所代表的数

所以我们只需要不断更新最近公共祖先，直到存储的下标超过了数据的个数（一般是n）就可以停止了，以下是代码：

void modify(int id,int k) {
	while (id<=n) {
		tree[id]+=k;
		id+=lowbit(id);
	}
	return;
}

3.线段树的查询和query函数

这一部分的代码可能在不同题目里含义不同（比如在【模板】树状数组 1 - 洛谷和【模板】树状数组 2 - 洛谷中就不一致，需要仔细思考以进行实际应用）。

再次观察图以及各个坐标代表的二进制，不难发现，其上一个大区间的位置为其坐标-lowbit(其坐标)，此时利用一个变量ans（sum也可以）记录答案就行了。以下是代码：

int query(int id) {
	int ans=0;
	while (id!=0) {
		ans+=tree[id];
		id-=lowbit(id);
	}
	return ans;
}

Part 4:总结

树状数组和线段树相比适用范围较窄，能够使用树状数组解决的问题线段树大部分也能解决。但是树状数组所占用的空间较少，时间复杂度最坏时才会达到O(nlog₂n)，这一点比线段树好。而且它代码量较少，简单题更偏向于使用树状数组，总之各有利弊。

喜欢就点个赞吧！