树状数组

树状数组

用途：

$①、修某个点的值——单点修改$

$②、求某一区间的前缀和——区间求和$

$时间复杂度O(lo{g}_{2}n)。$

---

示例

$顾名思义：以树的形状存储。$

$以16个数的区间为例(数组长度为17，下标从1开始)：$

$①、单点修改：$

$定义原数组A,树状数组tr,初始化为0，输入的过程实际上是一个单点修改操作。$

$每输入一个A[i],会影响树上tr下标为i+lowbit(i)的值,因此要更新tr[i+lowbit(i)]，直到i+lowbit(i)>元素个数n。$

$其中lowbit(x)指x二进制形式下最低位的1代表的二进制数的值。$

$$\begin{gathered} 例如上图中，在输入A[6]的值后，受影响的点是8和16，故要给节点8和16加上A[6]。 \\ lowbit(6)=lowbit(110)=10=2,tr[6+2]=tr[8]+=A[6],tr[8+8]=tr[16]+=A[6]。 \end{gathered}$$

$显然地，修改的时间复杂度为O(lo{g}_{2}n)$。

$②、区间求和$

$树上节点tr[i]=A[i]+\sum A[i-lowbit(i)],i+lowbit(i)>0，$

$求区间[l,r]的和:tr[r]-tr[l-1]。$

$求tr[i]的操作是O(lo{g}_{2}n)的，因此区间求和的时间复杂度也是O(lo{g}_{2}n)的。$

---

代码：

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

void add(int w,int p)//单点修改
{
    for(int i=p;i<=n;i+=lowbit(i)) tr[i]+=w;
}

int query(int x)//区间[1,x]的和
{
    int sum=0;
    for(int i=x;i;i-=lowbit(i)) sum+=tr[i];
    return sum;
}

---

例题：

题目：
给定 n 个数组成的一个数列，规定有两种操作，一是修改某个元素，二是求子数列 [a,b] 的连续和。

输入格式
第一行包含两个整数 n 和 m，分别表示数的个数和操作次数。

第二行包含 n 个整数，表示完整数列。

接下来 m 行，每行包含三个整数 k,a,b （k=0，表示求子数列[a,b]的和；k=1，表示第 a 个数加 b）。

数列从 1 开始计数。

输出格式
输出若干行数字，表示 k=0 时，对应的子数列 [a,b] 的连续和。

数据范围
1≤n≤100000,
1≤m≤100000，
1≤a≤b≤n
输入样例：
10 5
1 2 3 4 5 6 7 8 9 10
1 1 5
0 1 3
0 4 8
1 7 5
0 4 8
输出样例：
11
30
35

题意：

$$\begin{gathered} 输入数组长度n,操作数m,长度为n的数组，m个操作。 \\ k=0时输出区间[a,b]的和，k=1时把节点a的值加上b。 \end{gathered}$$

代码:

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#define ll long long
#define inf 0x7fffffff
using namespace std;
const int N=1e5+10;
int n,m;
int tr[N],a[N];

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

void add(int w,int p)
{
    for(int i=p;i<=n;i+=lowbit(i)) tr[i]+=w;
}

int query(int x)
{
    int sum=0;
    for(int i=x;i;i-=lowbit(i)) sum+=tr[i];
    return sum;
}
int main()
{
    cin>>n>>m;
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&a[i]);
        add(a[i],i);///建树,更新树上节点的值
    }

    while(m--)
    {
        int k,a,b;
        scanf("%d%d%d",&k,&a,&b);
        if(!k) printf("%d\n",query(b)-query(a-1));
        else add(b,a);///在位置a加上b，修改
    }

    return 0;
}