首页 > 其他分享 >树状数组

树状数组

时间:2022-09-03 11:02:43浏览次数:76  
标签:前缀 树状 int 个数 数组 data

前言:

设有一个包含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,个数,数组,data
From: https://www.cnblogs.com/kkk05/p/16583725.html

相关文章

  • [Leetcode 189]轮转数组
    Leetocde189轮转数组这题能被用做mid题是因为一题多解,其中基于双指针的轮状数组解法是比较难的1.使用新数组__直接把第i个元素移到第(i+k)%numsize位置,类似循环队列voi......
  • 树状数组学习笔记
    ​ 一:树状数组定义望文生义,树状数组就是用树形结构来模拟数组的一种数据结构。二:图解(纯手绘,难看勿喷)​编辑C表示从1-k的和,C[1]=a[1]C[2]=C[1]+a[2]C[3]=a[3]C[......
  • 数组:移除指定元素
    题目:给你一个数组nums 和一个值val,你需要原地移除所有数值等于 val 的元素,并返回移除后数组的新长度。不要使用额外的数组空间,你必须仅使用O(1)额外空间并原地......
  • react-native 类组件调用函数组件里面的方法
    父组件class类组件,关键代码我用红色部部分标记importReactfrom"react";import{View,Text}from"react-native";importHomeModalChilderfrom"../../compon......
  • java二维数组
    二维数组二维数组可以看成以数组为元素的数组。还可以有二维、三维、甚至更多维数组,但是实际开发中用的非常少。最多到二维数组(学习容器后,我们一般使用容器,二维数组用的......
  • 原生js处理数组数据小方法
    <ulid="ul"></ul><script>//假设这是一组数据,要渲染到页面页面vararr=[{msg:"nisjsiqi",img:"https://img1.baidu.com/it......
  • js数组方法
    修改器方法会改变自身的值push(element1,...,elementN):在数组的末尾增加一个或多个元素,并返回数组的新长度。pop():删除数组的最后一个元素,并返回这个元素(数组为空......
  • 循环遍历数组
    以数组的长度作为遍历数组/for循环遍历数组,while循环遍历数组;挨个打印称之为遍历数组1<script>2//打印出数组中的最大值3vararr=[12,89,11,0.8,178,96];4va......
  • 遍历数组元素
    1<!DOCTYPEhtml>2<html>3<head>4<metacharset="utf-8">5<title></title>6</head>7<body>8<script>9......
  • 求数组元素的和
    1<!DOCTYPEhtml>2<html>3<head>4<metacharset="utf-8">5<title></title>6</head>7<body>8<script>9......