树状数组其初衷是解决数据压缩里的累积频率(Cumulative Frequency)的计算问题,现多用于高效计算数列的前缀和, 区间和。它可以以 {\displaystyle O(\log n)} O(\log n)的时间得到任意前缀和 {\displaystyle \sum _{i=1}^{j}A[i],1<=j<=N} {\displaystyle \sum _{i=1}^{j}A[i],1<=j<=N},并同时支持在 {\displaystyle O(\log n)} O(\log n)时间内支持动态单点值的修改。空间复杂度 {\displaystyle O(n)} O(n)。
解决动态前缀和的数据结构
数组a1,a2,a3,a4…an
支持查询a1+a2+a3+…am的前缀和
也可以修改其中ai的元素,在进行查询前缀和,复杂度为o(nlogn)
步骤详解:
设节点编号为x,那么这个节点管辖的区间为2^k(其中k为x二进制末尾0的个数)个元素。因为这个区间最后一个元素必然为Ax
怎么理解呢?以上图为例:
c[6]=a5+a6 6换算成二进制为110,末尾有1个0,其管辖的区间为2^1个元素。
同理,c[8]=a1+a2+…a8,8换算成二进制为1000,末尾有3个0,其管辖的区间为2^ 3个元素。
查询操作:
见上图
比如询问前13项元素的值怎么做?
通过图不难发现 c[13]= c[13]+c[12]+c[8]
13换算成二进制为1101
1101 c[13]管辖2^0个元素,
1100 c[12]管辖2^2个元素
1000 c[8] 管辖2^3个元素
在最后一个1的位置减1
int query(int x){
int res=0;
while(x){
res+=tree[x];
x-=lowbit(x);
}
}
修改操作:
比如修改第6项元素会怎么样,
通过上图不难发现,修改c[6]会影响到c[6],c[8],c[16],也就是说所有覆盖c[6]的区间要进行修改
6换算成二进制为110
0110
1000
10000
在最后一个1的位置加1
void add(int x,int val){
while(x<=n){
tree[x]+=val;
x+=lowbit(x);
}
}
关键来了,怎样得到最后一个1在哪个位置呢?
lowbit预备函数,返回二进制下最后1个数在哪里?
int lowbit(int x){
return x&(-x);
}
比如12 为 1100
负数的补码为 按位取反在加1 也就是0011+1=0100
1100
& 0100
= 0100
可以算出12的最后一个1在哪个位置啦
完整版
int tree[300005],n;
int lowbit(int x){
return x&(-x);
}
int query(int x){
int res=0;
while(x){
res+=tree[x];
x-=lowbit(x);
}
}
void add(int x,int val){
while(x<=n){
tree[x]+=val;
x+=lowbit(x);
}
}
区间修改:
对区间[4,8]进行+10操作
只需要在4的位置上+10,在9的位置 -10就ok了
练习例题:POJ2352二维偏序问题
bzoj1878 ,2743,1452
cf round755 D