树状数组简单说明

树状数组其初衷是解决数据压缩里的累积频率(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

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值