树状数组
作者：互联网
前言：
设有一个包含n个数的数组a[ ] = { 9,2,5,6,3,12...... }，现要求前i个数的和值，即前缀和sum[ i ] = a[ 1 ]+a[ 2 ]+a[ 3 ]+...+a[ i ]（i = 1,2,3...n），可以怎样求？
首先想到的肯定是累加前i个数，但若对a[ i ]进行修改，则sum[ i ],sum[ i +1 ]...sum[ n ]都需要修改，最多需要修改n次，如果数据过大，则所需时间更长。因此，有人提出了树状数组，以便跟高效的求解前缀和与数值的修改。
那么，树状数组是如何巧妙实现的呢？
树状数组：
树状数组引入了分级管理的思想，设置一个管理小组，每个管理员管理一个或多个连续的元素。如下图：
管理数组是c[ ]，c[ i ]分别管理对应元素，如c[ 2 ]管理a[ 1 ]和a[ 2 ]，因为管理数组c[ ]是树状的，所以称为树状数组。
树状数组通过二进制划分管理空间，那么，c[ i ]的管理空间应如何划分呢？
1）区间长度：
若 i 的二进制表示末尾有连续 k 个0，则c[ i ]的区间长度为2k，即c[ i ]管理从a[ i ]起向前数2k个元素，则c[ i ]表示为a[ i-2k+1 ]+a[ i-2k+2 ]+a[ i-2k+3 ]+...+a[ i ]。如图：
例如 i =20，其二进制为10100，则区间长度为22=4，也就是其二进制最低位的1与其后面的0构成的二进制数值100，其十进制为4。
当然，这样写起来同样不方便，但在一些大佬的推导下得出了一个公式：
c[ i ]的区间长度为 i 的二进制取反（~）加1在进行与运算（&）。
下面我们来看 i =20时的例子：
大家会发现， i 取反加1的二进制与 -i 相等，因此可以得出公式：
c[ i ]的区间长度：lowbit(i) = ( -i )& i 。
2）前驱与后继：
直接前驱：c[ i ]的直接前驱为c[ i - lowbit(i) ]，即c[ i ]左侧紧邻的子树的根；
直接后继：c[ i ]的直接后继为c[ i +lowbit(i) ]，即c[ i ]的父节点；
前驱：c[ i ]左侧的所有子树的根；
后继：c[ i ]的所有祖先。
以c[ 6 ]为例：
所以，sum[ i ]的值为c[ i ]及其所有前驱的和。如：sum[ 7 ] = c[ 7 ] + c[ 6 ] + c[ 4 ]。
a[ i ]到a[ j ]的区间和为c[ j ] - c[ i-1 ]。
了解了这些以后，就可以开写代码啦~
首先，写好主函数：
1 int main()
2 {
3 int k;//共k组数据
4 cin >> k;//输入
5 while(k--)//循环k次
6 {
7 cin >> n >> m;//输入
8 for(int i=1; i<=n; i++)//循环n次输入
9 {
10 cin >> data[i];//先输入数据
11 lj(i,data[i]);//再建树
12 //也可以直接这么写
13 /*
14 int temp;//data[]数组是非必要的
15 cin >> temp;
16 lj(i,data[i]);
17 */
18 }
19 while(m--)//操作m次
20 {
21 int t,a,b;//操作指令，要操作的数，b
22 cin >> t >> a;//输入，若t==1或t==2，再输入b
23 if(t==1)//t==1，则第a个数加b
24 {
25 cin >> b;//输入b
26 lj(a,b);//累加
27 }
28 else if(t==2)//t==2，则求第a个数到第b个数的区间值
29 {
30 cin >> b;//输入b
31 cout << qjh(a-1,b) << endl;//求区间和
32 }
33 else//t==3，输出第1个数到第a个数的前缀和
34 {
35 cout << qzh(a) << endl;//输出
36 }
37 }
38 }
39 return 0;
40 }
接着，写计算区间长度的函数：
1 int lowbit(int i)//区间长度
2 {
3 return (-i)&i;//该操作为i取反加一再进行与运算
4 }
（一定要是int类型，要反回值的）
再把累加c[ ]数组的函数写好（我代码里的c[ ]数组写成sz[ ]了）：
1 void lj(int a,int b)//累加，构建树状数组（英语不好，拼音来凑）
2 {
3 for(int i=a; i<=n; i+=lowbit(i))//从第i个开始，i的所有后继加b
4 {
5 sz[i]+=b;//加b
6 }
7 }
然后就可以计算前缀和和区间值啦：
1 int qzh(int a)//前缀和
2 {
3 int s=0;//定义一个int变量
4 for(int i=a; i>0; i-=lowbit(i))//从i开始，i的所有前驱一一累加即为第一个数到i的前缀和
5 {
6 s+=sz[i];//累加
7 }
8 return s;//返回前缀和
9 }
10
11 int qjh(int a,int b)//区间和
12 {
13 return qzh(b)-qzh(a);//用第b个数的前缀和减去第a个数的前缀和即为第a个数到第b个数的区间和
14 }
然后就没了。
上代码~
代码：
1 #include<bits/stdc++.h>
2
3 using namespace std;
4
5 const int maxn=10005;//定义一个大点的数
6
7 int n,m;//每组数据的数字数，操作数
8 int data[maxn],sz[maxn];//数据数组，树状数组
9
10 int lowbit(int i)//区间长度
11 {
12 return (-i)&i;//该操作为i取反加一再进行与运算
13 }
14
15 void lj(int a,int b)//累加，构建树状数组（英语不好，拼音来凑）
16 {
17 for(int i=a; i<=n; i+=lowbit(i))//从第i个开始，i的所有后继加b
18 {
19 sz[i]+=b;//加b
20 }
21 }
22
23 int qzh(int a)//前缀和
24 {
25 int s=0;//定义一个int变量
26 for(int i=a; i>0; i-=lowbit(i))//从i开始，i的所有前驱一一累加即为第一个数到i的前缀和
27 {
28 s+=sz[i];//累加
29 }
30 return s;//返回前缀和
31 }
32
33 int qjh(int a,int b)//区间和
34 {
35 return qzh(b)-qzh(a);//用第b个数的前缀和减去第a个数的前缀和即为第a个数到第b个数的区间和
36 }
37
38 int main()
39 {
40 int k;//共k组数据
41 cin >> k;//输入
42 while(k--)//循环k次
43 {
44 cin >> n >> m;//输入
45 for(int i=1; i<=n; i++)//循环n次输入
46 {
47 cin >> data[i];//先输入数据
48 lj(i,data[i]);//再建树
49 //也可以直接这么写
50 /*
51 int temp;//data[]数组是非必要的
52 cin >> temp;
53 lj(i,data[i]);
54 */
55 }
56 while(m--)//操作m次
57 {
58 int t,a,b;//操作指令，要操作的数，b
59 cin >> t >> a;//输入，若t==1或t==2，再输入b
60 if(t==1)//t==1，则第a个数加b
61 {
62 cin >> b;//输入b
63 lj(a,b);//累加
64 }
65 else if(t==2)//t==2，则求第a个数到第b个数的区间值
66 {
67 cin >> b;//输入b
68 cout << qjh(a-1,b) << endl;//求区间和
69 }
70 else//t==3，输出第1个数到第a个数的前缀和
71 {
72 cout << qzh(a) << endl;//输出
73 }
74 }
75 }
76 return 0;
77 }
练习：
再来两道练习吧~
1.P3374
代码：
这个题写起来极其简单，完全就是上边树状数组的默写，尽量不看代码默写。
1 #include<bits/stdc++.h>
2
3 using namespace std;
4
5 const int maxn=500005;//根据题目定义一个大点的数
6
7 int n,m;//每组数据的数字数，操作数
8 int data[maxn],sz[maxn];//数据数组，树状数组
9
10 int lowbit(int i)//区间长度
11 {
12 return (-i)&i;//该操作为i取反加一再进行与运算
13 }
14
15 void lj(int a,int b)//累加，构建树状数组
16 {
17 for(int i=a; i<=n; i+=lowbit(i))//从第i个开始，i的所有后继加b
18 {
19 sz[i]+=b;//加b
20 }
21 }
22
23 int qzh(int a)//前缀和
24 {
25 int s=0;//定义一个int变量
26 for(int i=a; i>0; i-=lowbit(i))//从i开始，i的所有前驱一一累加即为第一个数到i的前缀和
27 {
28 s+=sz[i];//累加
29 }
30 return s;//返回前缀和
31 }
32
33 int qjh(int a,int b)//区间和
34 {
35 return qzh(b)-qzh(a);//用第b个数的前缀和减去第a个数的前缀和即为第a个数到第b个数的区间和
36 }
37
38 int main()
39 {
40 cin >> n >> m;//输入
41 for(int i=1; i<=n; i++)//循环n次输入
42 {
43 cin >> data[i];//先输入数据
44 lj(i,data[i]);//再建树
45 }
46 while(m--)//操作m次
47 {
48 int t,a,b;//操作指令，要操作的数，b
49 cin >> t >> a >> b;//输入
50 if(t==1)//t==1，则第a个数加b
51 {
52 lj(a,b);//累加
53 }
54 else//t==2，则求第a个数到第b个数的区间和
55 {
56 cout << qjh(a-1,b) << endl;//求区间和
57 }
58 }
59 return 0;
60 }
2.P3368
讲解：
这个题如果直接在区间内加k的话会超时，所以要引入一个新的东西——差（cha，一声）分。
简单地举个例子：
设数组a[ ] = {1,6,8,5,10}，那么差分数组b[ ] = {1,5,2,-3,5}，即差分数组b[ i ] = 数据数组a[ i ] - a[ i -1 ]，而由b[ i ]还原a[ i ]也不难，a[ i ] = b[ i ] + b[ i -1 ] + ...... + b[ 1 ]（可以证）。
如果让a[ 2 ]到a[ 4 ]间都加2，数据数组a变为a[ ] = {1,8,10,7,10}，差分数组b变为b[ ] = {1,7,2,-3,3}，可以发现，只有b[ 2 ]和b[ 5 ]变了，这样就只用修改两个数。
所以，这个题中需要把树状数组由数值变为数值间的差分进行计算，节省时间。
代码：
1 #include<bits/stdc++.h>
2
3 using namespace std;
4
5 const int maxn=500005;//根据题目定义一个大点的数
6
7 int n,m;//每组数据的数字数，操作数
8 int data[maxn],sz[maxn];//数据数组，树状数组（且是差分数组）
9
10 int lowbit(int i)//区间长度
11 {
12 return (-i)&i;//该操作为i取反加一再进行与运算
13 }
14
15 void lj(int a,int b)//累加，构建差分树状数组
16 {
17 for(int i=a; i<=n; i+=lowbit(i))//从第i个开始，i的所有后继加b
18 {
19 sz[i]+=b;//加b
20 }
21 }
22
23 int qzh(int a)//前缀和（可通过证明验证差分树状数组前i个的前缀和为第i个数的值）
24 {
25 int s=0;//定义一个int变量
26 for(int i=a; i>0; i-=lowbit(i))//从i开始，i的所有前驱一一累加即为第一个数到i的前缀和
27 {
28 s+=sz[i];//累加
29 }
30 return s;//所以这里返回的是树状数组的前缀和，数据数组的第i个数
31 }
32
33 int qjh(int a,int b)//区间和（写上没用，删了吧）
34 {
35 return qzh(b)-qzh(a);//用第b个数的前缀和减去第a个数的前缀和即为第a个数到第b个数的区间和
36 }
37
38 int main()
39 {
40 cin >> n >> m;//输入
41 for(int i=1; i<=n; i++)//循环n次输入
42 {
43 cin >> data[i];//先输入数据
44 lj(i,data[i]-data[i-1]);//再建树（因为是从data[1]开始，所以data[1-1]一定为0，sz[1]==data[1]）
45 }
46 while(m--)//操作m次
47 {;
48 int a,b;//操作指令
49 cin >> a >> b;//输入
50 if(a==1)//a==1，则第b个数到第c个数加d
51 {
52 int c,d;//
53 cin >> c >> d;//
54 lj(b,d);//
55 lj(c+1,0-d);//
56
57 }
58 else//a==2，输出第b个数的值（因为前面有部分数已经加上了别的数，所以data[]数组与修改后的sz[]数组不同步，无法继续使用）
59 {
60 cout << qzh(b) << endl;//输出第b个数
61 }
62 }
63 return 0;
64 }
就没有啦，谢谢观看~
标签：前缀,树状,int,个数,cin,数组,data 来源： https://www.cnblogs.com/kkk05/p/16583725.html