树状数组是一个查询和修改复杂度都为log(n)的数据结构,用于查询任意一个区间的所有元素之和。
用途:区间和查询,单点修改
树状数组如图建立,那么树状数组怎么就方便了呢?如果我们想要知道14的前缀和,那么我们只用知道树状数组13-14这个位置的值和9-12这个位置的值,还有1-8这个位置的值,我们只用加三次就可以知道答案,比起原来的一个一个求和操作次数是不是更少了呢。
令这棵树的节点编号为1,2,....,n
那么d[1]=a[1]
d[2]=a[1]+a[2]
d[3]=a[3]
d[4]=a[4]+a[3]+a[2]+a[1]
d[5]=a[5]
d[6]=a[5]+a[6]
..........
设节点编号为x,那么这个区间管辖的元素为2^k个,(k为x转换为二进制末尾零的个数)。
x=1,1的二进制为0001,管理的元素为2^0=1个
x=2,2的二进制为0010,管理的元素为2^1=2个
x=4,4的二进制为0100,管理的元素为2^2=4个
那么我们怎么找到这个k呢?可以利用补码
其中-x是对x求补码,比如x=12,二进制为1100,补码为各个位取反+1 取反0011+0001=0100 最终0100为补码。
1100&0100=0100=2^2=4最后返回的值为4用这个就可以知道d[x]管理的元素个数啦~
int lowbit(int x)
{
return x&(-x);
}
如果我们要询问13这个位置的前缀和
13的二进制为1101 根据它的二进制可以拆分成三部分1101 1100 1000 我们发现每次都是将其末尾的1变成0
1101对应的数字是13,1100对应的数字是12,1000对应的数字是8,13的二进制1101 2^0=1 1的二进制为0001 1101-0001=1100
对应数字为12 同理1100-0100(2^2的二进制)=1000(8的二进制)
我们再看下那个图,13的前缀和=d[13]+d[12]+d[8] (自上而下)
再根据前面我们找到的规律d[x]所管辖的元素个数为x转化为二进制以后的末尾0的个数,d[13] 1101 2^0=1个元素 d[12] 1100 2^2=4个元素 d[8] 1000 2^3=8个元素 而2^0+2^2+2^3=1+4+8=13恰好为13
(这里求的是1-13这个位置的和,那么如果我们要求一个区间[5,8]的和怎么办呢?只要求query(8)-query(4)即可,就是1-8区间的和减去1-4这个区间的和,剩下的不就是5-8这个区间的和了吗)
int query(int x) //拿x=13当例子
{
int sum=0;
while(x)
{
sum+=d[x]; //sum+=d[13]; sum+=d[12]; sum+=d[8];
x-=lowbit(x); //13-=lowbit(13)=12; 12-=lowbit(12)=8; 8-=lowbit(8)=0;
}
return sum;
}
那如果是修改操作呢,假设我们修改6这个位置的值,那么我们就需要把覆盖这个值的节点的值都修改了,如上图,6这个节点的值就需要修改,还有8这个节点的值,16这个节点的值(自下而上)。
如果我们要修改(增加或减少)6这个位置的值,6的二进制为0110 2^1=2个元素 ,6+2=8 ,8的二进制1000,2^3=8 ,8+8=16 ,16的二进制为10000 .我们再看6 二进制0110 管理2个元素 2的二进制0010 0110+0010=1000 同理 1000+1000=10000(逢二进一)
void update(int x,int v)
{
while(x<=n)
{
d[x]+=v;
x+=lowbit(x);
}
}