C++树状数组

树状数组

树状数组一般用于对数组求前缀和及修改某个下标的数据时。当解决这类问题时，每次求前缀和都循环累加会导致求前缀和的速度过慢，而使用前缀和数组会导致修改数据的速度过慢。而树状数组每个操作时间复杂度都是O(logn)。

原理

求和

求和时如图所示。

公式如下：

c [ i ] = a [ i - 2 ^ k + 1 ] + a [ i - 2 ^ k + 2 ] + ... + a [ i ]

其中k为i的二进制中末尾有几个0

例如：

想要求c[3]，此时i=3，3的二进制为11，末尾没有0，所以k=0，2^k=1。

于是c[3]=a[3]

若想求c[4]，此时i=4，4的二进制为100，末尾有两个0，所以k=2,2^k=4。

于是c[4]=a[1]+a[2]+a[3]+a[4]

2^k可以用lowbit(i)，或i&(-i)

修改

修改某个点的数据时，也需要更新往后的数据。

当c[i]被修改时，后面的c[i+lowbit(i)]也要修改，一直向后推，直到修改至最末尾的点。

for ( ; i <= n ; i += lowbit ( i ) ) 

例如：

想修改a[3],首先c[3]修改，然后根据公式得c[4]需要修改。直接将c[4]增加a[3]的变化量。接着继续按照公式得到需要修改c[8]。此时已经到数组最后一个点，无需继续修改。

此种修改方式时间复杂度为O(logn)

查询

想要查询a[1]~a[x]区间的和的时候，可以从c[x]开始累加，接着将下标减去lowbit(x)，以此类推，直到x-lowbit(x)<0。

for(int i=x;i>=0;i-=lowbit(i))s+=c[i];

例如：

想求a[1]~a[6]的和，首先将sum加上c[6],然后根据公式得到还需加上c[4]，接着就不需要再加了，此时sum就是求和的答案。

题目

BIT-1

题目描述

给定长度为n的序列a，进行q次操作，每次操作有两种

1 i  x 将第i个变为x

2 l r 求l-r区间的和

思路

首先要写lowbit函数

ll lowbit(ll x){
    return x&(-x);
}

接着建立树状数组c维护1-x的前缀和，并根据树状数组的原理写出update函数

void update(ll x,ll y){
    for(int i=x;i<=n;i+=lowbit(i))c[i]+=y;
}

然后是查询区间和。写一个函数，将区间头和区间末尾的前缀和相减即可

ll qq(ll l,ll r){
    ll lll=0,rr=0;
    for(int i=r;i;i-=lowbit(i))rr+=c[i];
    for(int i=l-1;i;i-=lowbit(i))lll+=c[i];
    return rr-lll;
}

AC代码

BIT-2

题目描述

 

给定长度为n的序列a，进行q次操作，每次操作有两种

1 l r x 将l-r区间整体增加x

2 k 求第k项

思路

这道题首先维护一个差分数组b，因为区间整体增加，所以区间内部相对大小没有变化，只有头和尾后的一个元素需要修改差分数组。

于是可以写出update函数

void update(ll x,ll y){
    for(int i=x;i<=n;i+=lowbit(i))b[i]+=y;
    for(int i=x;i<=n;i+=lowbit(i))b[i]-=y;
}

因为这道题查询只查询一个元素，所以可以写出查询的函数

ll qq(ll x){
    ll ans=0;
    for(int i=x;i;i-=lowbit(i))ans+=b[i];
    return ans;
}

AC代码

BIT-3

题目描述

 

给定长度为n的序列a，进行q次操作，每次操作有两种

1 l r x 将l-r区间整体增加x

2 l r 求l-r和

思路

用差分数组d存储a[i]-a[i-1]。

可以推出求a前缀和的式子

所以需要维护两个树状数组，一个维护d前缀和，一个维护d[i]*[i]前缀和

本道题还没有通过