树状数组入门
第一次听见树状数组这个名词,我就觉得好高端,但是熟悉了之后,才发现世界上居然还有如此平易近人的数据结构,树状数组各种好,简短高效常数小,虽然应用范围明显小于线段树,但是在两者都可以解决的问题上,树状数组有明显的优势,讲了这么多,它到底是什么?可以解决什么问题?
先解决第二个问题,树状数组可以干什么,它可以对一个数列进行区间、单点修改,可以求某一个数或区间和,还有很多别的问题,都可以由这四个操作以O(nlogn)的效率完成。至于更高端的操作……我们只讲入门:)
而第一个问题,我们从一张图开始(网上找到,如有雷同,不要见怪):
相信大家在看见图时就明白树状数组是哪一部分了,就是红色的部分,它的名字正是由它的形状而来,为什么它要长成这样?这样做为什么高效呢?观察它的结构,再结合效率中的log,相信已经有人发现了它的奥秘,没错!就是二分,数字二分。
我们从构建它的结构出发。a数组是已知的一个数列,c就是树状数组,它所存储的,是我们对它所管理或控制的范围内ai的操作,原理类似线段树的lazy tag,那么当我们修改一个数ai的值时,我们就必须修改它在c的所有祖先(类比树结构),比如修改图中的a3,我们要修改c3,c4,c8,而这个操作,我们可以通过一个叫lowbit的东西来完成,lowbit指的是一个数的二进制的最低位1表示的数,因为是最低位的1,所以lowbit必然为2^k,这点可以由2^k的二进制特点:10000……得到,而如果你在修改时每一次+=lowbit,就可以以近乎倍增的效率向上直到到达边界,把所有祖先修改,而不是修改之后的所有数据,就是log的效率,这个操作的代码如下:
void add(int t,int y)
{
while(t<=n)
{
tre[t]+=y;
t+=lowbit(t);
}
}
是不是很粗暴,如果你不放心,大可以写一个程序让一个数+=lowbit一直加下去看看,说的直白一点,+=lowbit事实上就是二进制中的进位操作。
而另一个操作,求前缀和,事实上就是将t+=lowbit变为-=直到0,你会惊奇的发现,这样的话,可以涵盖t之前的所有的a而不重复,达到了我们的目的
int getsum(int h)
{
int sum=0;
while(h)
{
sum+=tre[h];
h-=lowbit(h);
}
return sum;
}
给出两道模板题:洛谷 P3368 P3374
而树状数组的另一个重要操作,求逆序对及其必备优化,将在我的下一篇博客中简述,对树状数组做个总结,它其实就是利用二分与二进制,来实现对于一个序列的O(logn)的查询、维护、更改……二进制的思想在其中发挥了重要作用,强烈建议好好学一下,我也刚学不久,可能有些地方不太完善,请多包容,谢谢。