学习笔记:树状数组

树状数组是一个查询和修改复杂度都为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);
    }
}

 

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值