树状数组介绍

---

一、树状数组是什么？

树状数组是一种支持 单点修改 和 区间查询 的，代码量小的数据结构。

二、工作原理

1.初步感受

其中a数组是原始数据数组，c数组可以看成管理a局部和的数组。例如
$c_2$ 管辖的是 $a[1\dots 2]$；
$c_4$ 管辖的是 $a[1\dots 4]$；
$c_6$ 管辖的是 $a[5\dots 6]$；
$c_8$ 管辖的是 $a[1\dots 8]$；
剩下的 c[x] 管辖的都是 a[x] 自己（可以看做 $a[x\dots x]$ 的长度为 1 的小区间）。

2.求和举例

例如
$a[1\dots 7]$的前缀和为$a_1+a_2+a_3+a_4+a_5+a_6+a_7$ = $c_4+c_6+c_7$
$a[4\dots 7]$的前缀和为$a[1\dots 7]-a[1\dots 3]$ = $c_4+c_6+c_7-c_2-c_3$

3.管辖区间

树状数组中，规定$c[x]$管辖的区间长度为 $2^k$，其中：

设二进制最低位为第 0 位，则 $k$ 恰好为 $x$ 二进制表示中，最低位的 1 所在的二进制位数；
$2^k$（$c[x]$ 的管辖区间长度）恰好为 $x$ 二进制表示中，最低位的 1 以及后面所有 0 组成的数

举个栗子
$c_{88}$ 管辖的是哪个区间？
因为 $88_{(10)}$=$01011000_{(2)}$，其二进制最低位的 1 以及后面的 0 组成的二进制是 1000，即 8，所以 $c_{88}$ 管辖 8 个 $a$ 数组中的元素。因此，$c_{88}$ 代表 $a[81\dots 88]$ 的区间信息。

我们记 $x$ 二进制最低位 1 以及后面的 0 组成的数为 $\mathrm{lowbit}(x)$，那么 $c[x]$ 管辖的区间就是 $[x-\mathrm{lowbit}(x)+1,x]$。

注意这里的lowbit(x) = x & -x

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

4.区间查询

其实任何一个区间查询都可以这么做：查询 $a[l\dots r]$ 的和，就是 $a[1\dots r]$ 的和减去 $a[1\dots l-1]$ 的和，从而把区间问题转化为前缀问题，更方便处理。

那么如何做前缀查询，回顾一下查询 $a[1\dots 7]$ 的过程：

从 $c_7$ 往前跳，发现 $c_7$ 只管辖 $a_7$ 这个元素；然后找 $c_6$，发现 $c_6$ 管辖的是$a[5\dots 6]$，然后跳到 $c_4$，发现 $c_4$ 管辖的是 $a[1\dots 4]$这些元素，然后再试图跳到 $c_0$，但事实上 $c_0$ 不存在，不跳了。

我们刚刚找到的 $c$ 是 $c_7,c_6,c_4$，事实上这就是 $a[1\dots 7]$拆分出的三个小区间，合并一下，答案是 $c_7+c_6+c_4$。

我们可以写出查询 $a[1\dots x]$ 的过程：

从 $c[x]$ 开始往前跳，有 $c[x]$ 管辖 $a[x-\mathrm{lowbit}(x)+1\dots x]$；
令 $x\leftarrow x-\mathrm{lowbit}(x)$，如果 $x=0$ 说明已经跳到尽头了，终止循环；否则回到第一步。将跳到的 $c$ 合并。

int getsum(int x) {  // a[1]..a[x]的和
  int ans = 0;
  while (x > 0) {
    ans = ans + c[x];
    x = x - lowbit(x);
  }
  return ans;
}

5.单点修改

现在来考虑如何单点修改 $a[x]$。我们的目标是快速正确地维护 $c$ 数组。为保证效率，我们只需遍历并修改管辖了 $a[x]$ 的所有$c[y]$，因为其他的 $c$ 显然没有发生变化。注意$c[y]$必定包含$c[x]$。

设 $n$ 表示 $a$ 的大小，不难写出单点修改 $a[x]$ 的过程：

初始令 $x^{\prime }=x$。
修改 $c[x^{\prime }]$。
令 $x^{\prime }\leftarrow x^{\prime }+\mathrm{lowbit}(x^{\prime })$，如果 $x^{\prime }>n$ 说明已经跳到尽头了，终止循环；否则回到第二步。

void add(int x, int k) {
  while (x <= n) {  // 不能越界
    c[x] = c[x] + k;
    x = x + lowbit(x);
  }
}

总结

树状数组有很多变种，这里不一一赘述，如有兴趣，可以参考https://oi.wiki/ds/fenwick/