树状数组1

1 介绍

给出一个长为 $n$ 的数组 $a$，进行若干次几下操作：

1. 改变 $a_x$ 的值
2. 求 ${\sum }_{i=1}^x{a}_i$ 并输出

树状数组是一种树形结构的数组，以二叉树的方式进行存储。它的基本操作有：改变元素的值、求区间前缀和。

树状数组的优越性在于其进行操作的复杂度都是 $O(nlogn)$.

2 前置内容

求一个数的二进制从右往左第一个 $\text{1}$ 与后面末尾 $\text{0}$ 组成的数 lowbit().

例如：$lowbit(4)=4,lowbit(7)=1$.

实现方法：

void lowbit(int n) {
    return n & (-n);
}
//other examples
//return n & (~n + 1);
//return n & (n ^ (n - 1));

3 实现

以一个长度为 $8$ 的数组 $a$ 举例：

现将数组 $a$ 的所有元素都放到二叉树的叶子结点上，然后进行操作，得到树状数组 $c$.

$c_1=a_1(1{)}_{10}=(0001{)}_2$

$c_2=a_1+a_2(2{)}_{10}=(0010{)}_2$

$c_3=a_3(3{)}_{10}=(0011{)}_2$

$c4=a_1+a_2+a_3+a_4(4{)}_{10}=(0100{)}_2$

$c_5=a_5(5{)}_{10}=(0101{)}_2$

$c_6=a_5+a_6(6{)}_{10}=(0110{)}_2$

$c_7=a_7(7{)}_{10}=(0111{)}_2$

$c_8=a_1+a_2+a_3+a_4+a_5+a_6+a_7+a_8(8{)}_{10}=(1000{)}_2$

二叉树如下：

我们可以发现：

从下往上第 $i$ 层的编号二进制的末尾都是 $(i-1)$ 个 $\text{0}$，也就是从右往左第一个 $\text{1}$ 所在第 $i$ 位。

如果要改变某个元素的值（$a_x+y$），如 $a_3$ ，那么同样要改动 $c_3,c_4,c_8$ 的值。观察这三个元素下标的规律，我们可以发现：

$3+lowbit(3)=4，4+lowbit(4)=8$

可得代码：

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

如果要求 ${\sum }_{i=1}^x{a}_i$，如求从 $a_1$ 到 $a_7$ 的和，那么可以通过 $c_4+c_6+c_7$ 得到。观察这三个元素下标的规律，我们可以发现：

$6=7-lowbit(7),4=6-lowbit(6)$

可得代码：

int count(int x) {
    int res = 0;
    for (int i = x;i >= 1; i -= lowbit(i)) res += c[i];
    return res;
}

如果要求 ${\sum }_{i=l}^r{a}_i$，利用前缀和，${\sum }_{i=1}^r{a}_i-{\sum }_{i=1}^{l-1}{a}_i$ ， count(r) - count(l)即可。

4 离散化

存储大数的方法，例如：

1,10000000,99999999,12345,64678

将它们排序得到：

1,12345,64678,10000000,99999999

对他们标号，依次对应：

1,2,3,4,5

可以利用 map 来保存编号。

代码如下：

for (int i = 1; i <= n; i++)
	b[i] = a[i];
sort(b + 1, b + n + 1);
for (int i = 1; i <= n; i++) 
    mp[b[i]] = i;