数据结构——树状数组

首先，要了解一下功能比他多，码量比他要多的线段树

了解了线段树，我们开始讲解树状数组。

树状数组的概念

树状数组是只维护区间和的，他本来是一颗线段树，然后可以发现，每一个结点的左儿子都是没有用的，于是就变成了一颗结点数为n的数，如下图：

1---2---3---4---5---6---7---8

1---2---3---4

1---2                    5---6

1            3             5            7

存树不需要用链表，可以用数组，因为每个结点的结束地点都是不一样的，只要结点的结尾是多少，他的下标就是多少。

lowbit函数

这个是学习树状数组时必须知道的东西，他帮我们找区间。

lowbit，顾名思义，就是一个数字最小的为1的二进制位。

比如说9，他的二进制是1001，最小一位是，所以9的lowbit就是也就是1。

取lowbit的方式很简单，首先要知道一点，一个数变成负数，二进制码的最后几位，也就是数字位，会取反并加1，1001反一下就是0110，再加1就是0111这时，1001和0111只有第一位是一样的，所以只要做&运算就可以求出lowbit。

lowbit的代码：

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

修改、插入

观察上面的一棵树，可以发现，如果要修改，就可以先找到叶子，再不断加lowbit加出所有包含他的结点，如果要查询，就可以先找到要查到前缀和到末项不断减lowbit，减出整个前缀区间。

于是，加法就变成了把结点x到n中每个x+lowbit加得出来的结点都加num，求前缀就变成了把结点x到1之间每个x-lowbit减得出来的结点都加起来，代码也十分简单。

修改：

void add(int x,ll val){
    for (;x<=n;x+=lowbit(x)){
        c[x]+=val;
    }
}

查询：

ll sum(int x){
    ll res=0;
    for (;x;x-=lowbit(x)){
        res+=c[x];
    }
    return res;
}

注意：在这里，c只维护新加上来的量，原来的量的区间值要在输入时求出前缀和才行，像这样

for (int i=1;i<=n;i++){
    scanf("%lld",&a[i]);
    a[i]+=a[i-1];
}

然后sum函数返回的是前缀和，要求区间和要用1～r的和减掉1～l-1的值才行。

完整代码

#include <iostream>
using namespace std;
typedef long long ll;
ll a[1000005],c[1000005];
int n,q;
int lowbit(int x){
    return x&-x;
}
void add(int x,ll val){
    for (;x<=n;x+=lowbit(x)){
        c[x]+=val;
    }
}
ll sum(int x){
    ll res=0;
    for (;x;x-=lowbit(x)){
        res+=c[x];
    }
    return res;
}
int main() {
    scanf("%d%d",&n,&q);
    for (int i=1;i<=n;i++){
        scanf("%lld",&a[i]);
        a[i]+=a[i-1];
    }
    while (q--){
        int type;
        scanf("%d",&type);
        if (type==1){
            int i,x;
            scanf("%d%d",&i,&x);
            add(i,x);
        }
        else{
            int l,r;
            scanf("%d%d",&l,&r);
            printf("%lld\n",a[r]-a[l-1]+sum(r)-sum(l-1));
        }
    }
    return 0;
}

例题选做

单修区查        区修单查              区修区查​​​​​​​