树状数组

最新推荐文章于 2023-12-27 11:52:03 发布
最新推荐文章于 2023-12-27 11:52:03 发布
阅读量689 收藏 1
点赞数
分类专栏： 树状数组&线段树&字典树... 文章标签： c zk ini

我们知道要是一个数组的和C[i]=A[1]+A[2]+.......+A[i],那么我更改其中任意一项值A[i],那么C[i],C[i+1],C[i+2]....C[n]都会随之变化，所以我们要调整C[]当n非常大时就需要很长的时间，必然会导致超时。这样就可以引入树状数组，它的修改和求和的复杂度为nlogn.

树状数组是一种数组结构，能够高效地获取数组中连续n个数的和。

下面这张图是基本上大多数博客都有的一张图，主要为了让我们更容易理解树状数组。

红色的C[i]表示树状数组，且C[i] = A[i–2^k+ 1] +A[i-2^k+2] … + A[i] (i>=1)

其中，k为i在二进制下末尾0的个数，(可以利用位运算2^k=i&(i^(i-1))=i&(-i))，则我们称C为树状数组。由树状数组的定义可知：

C[1]=A[1]；   C[2]=A[1]+A[2]=C[1]+A[2]；  C[3]=A[3]； C[4]=A[1]+A[2]+A[3]+A[4]=C[2]+C[3]+A[4]；

C[5]=A[5]； C[6]=A[5]+A[6]=C[5]+C[6]； C[7]=A[7]； C[8]=...=C[4]+C[6]+C[7]+A[8];

由图可知，k为这颗树的高度，且最高不会超过logn，这个时候我们改变A[i]的值后可以由C[i]往根节点上溯，调整这条路上的C[]即可。比方说改变A[1]，即pos=1的值，需要更新的值为C[1],C[2],C[4],C[8],位置变化的规律为：pos+=LowBit(pos)，复杂度为logn。模板为：

[cpp]  view plain copy print ?
  1. #define LowBit(num) num&(-num)  
  2. void Add(int pos,int value)  
  3. {   for(int i=pos;i<MAX;i+=LowBit(i))  
  4.         C[i]+=value;  
  5. }  

要求统计某一个位置num的前num项和时：只要找到num前的所有的最大子树，然后把其根节点的C[]加起来即可。如：要求前5项的和，用Sum(5)表示，则Sum(5)=C[5]+C[4]，需要加起来的C[i]中的i也有规律可循，i-=LowBit(i),模板为：

[cpp]  view plain copy print ?
  1. int Add(int num)  
  2. {   int sum=0;  
  3.     for(int i=num;i>0;i-=LowBit(i))  
  4.         sum+=C[i];  
  5.     return sum;      
  6. }  

树状数组能快速求任意区间的和：A[i] + A[i+1] + … + A[j]，设sum(k) = A[1]+A[2]+…+A[k]，则A[i] + A[i+1] + … + A[j] = sum(j)-sum(i-1)。

用法1：树状数组---插点问线，当只有某一点的数值变化，而查找的区间为一段时用这个方法。

直接上题目吧，例题1:HDU 1166(敌兵布阵)直接用暴力的话肯定会超时。

[cpp]  view plain copy print ?
  1. #include<iostream>  
  2. #include<cstdio>  
  3. #include<cstring>  
  4. using namespace std;  
  5. const int MAX=1000010;  
  6. #define CLR(arr,val) memset(arr,val,sizeof(arr))  
  7. int n,total,value,C[MAX];  
  8. char Find[10];  
  9. int LowBit(int num)//二进制后面为0的个数  
  10. {   return num&(-num);  
  11. }  
  12. void Add(int pos,int val)//将pos位置的值+val  
  13. {   while(pos<=n)  
  14.     {   C[pos]+=val;//则每个位置的C[]都要改变，看下图有哪些C[]改变了，比方我把A[4]+val,变化的C[]有C[4],C[8]..变化,C[5],C[6]..并未变化  
  15.         pos+=LowBit(pos);//求出下一个要改变的位置，回溯到根节点  
  16.     }  
  17. }  
  18. int Sum(int num)//求前num项的和  
  19. {   int sum=0;  
  20.     while(num>0)  
  21.     {   sum+=C[num];  
  22.         num-=LowBit(num);//找到num前的所有最大子树  
  23.     }  
  24.     return sum;  
  25. }  
  26. int main()  
  27. {   int sum=1;  
  28.     scanf("%d",&total);  
  29.     while(total--)  
  30.     {   CLR(C,0);//要记得初始化   
  31.         printf("Case %d:\n",sum++);   
  32.         scanf("%d",&n);  
  33.         for(int i=1;i<=n;i++)//要从1开始!!!   
  34.         {   scanf("%d",&value);  
  35.             Add(i,value);  
  36.         }  
  37.             
  38.         while(scanf("%s",Find))  
  39.         {   int End,Start;  
  40.             if(Find[0]=='E') break;  
  41.             scanf("%d%d",&Start,&End);   
  42.             if(Find[0]=='A') Add(Start,End);  
  43.             else if(Find[0]=='S') Add(Start,-End);  
  44.             else printf("%d\n",Sum(End)-Sum(Start-1));   
  45.         }  
  46.     }  
  47.     return 0;  
  48. }  

和这个题目一样的是：NYOJ 116(士兵杀敌)~~~~~~
用法2：插线问点，当有某一段的数值都有变化时，而查询只有一个位置时用插线问点。

[cpp]  view plain copy print ?
  1. #include<iostream>  
  2. #include<cstring>  
  3. #include<cstdio>  
  4. using namespace std;  
  5. const int MAX=1000010;  
  6. #define CLR(arr,val) memset(arr,val,sizeof(arr))  
  7. int n,m,num,C[MAX];  
  8. char Find[10];  
  9. int LowBit(int num)  
  10. {   return num&(-num);  
  11. }  
  12. void Add(int pos,int val)//前n项每项增加val   
  13. {   while(pos>0)  
  14.     {   C[pos]+=val;  
  15.         pos-=LowBit(pos);      
  16.     }  
  17. }  
  18. int Sum(int pos)  
  19. {   int sum=0;  
  20.     while(pos<=n)  
  21.     {   sum+=C[pos];  
  22.         pos+=LowBit(pos);  
  23.     }  
  24.     return sum;  
  25. }  
  26. int main()  
  27. {   CLR(C,0);  
  28.     scanf("%d%d",&n,&m);  
  29.     for(int i=1;i<=n;i++)  
  30.     {   scanf("%s",Find);  
  31.         if(Find[0]=='A')  
  32.         {   int Start,End;  
  33.             scanf("%d%d%d",&Start,&End,&num);  
  34.             /****下面两行Nice****/  
  35.             Add(Start-1,-num);  
  36.             Add(End,num);  
  37.             /********************/  
  38.         }  
  39.         else   
  40.         {   scanf("%d",&num);  
  41.             printf("%d\n",Sum(num));  
  42.         }  
  43.     }  
  44.     return 0;  
  45. }  

用法3：多维树状数组，增加模板：

[cpp]  view plain copy print ?
  1. #define LowBit(num) num&(-num)  
  2. void Add(int x,int y,....,int val)  
  3. {   for(int i=x;i<MAX;i+=LowBit(i))  
  4.         for(int j=y;j<MAX;j+=LowBit(j))  
  5.             .......//是几维的就几层  
  6.             C[i][j][...]+=val;  
  7. }  
  8. void Sum(int x,int y,.....)  
  9. {   int sum=0;  
  10.     for(int i=x;i>0;i-=LowBit(i))  
  11.         for(int j=y;j>0;j-=LowBit(j))  
  12.             ......  
  13.             sum+=C[i][j][...];  
  14.     return sum;          
  15. }  

例题3：HDU 1892(See you~~)，代码：

[cpp]  view plain copy print ?
  1. #include<iostream>  
  2. #include<cstring>  
  3. #include<cstdio>  
  4. using namespace std;  
  5. const int MAX=1010;  
  6. #define min(a,b) a<b?a:b   
  7. #define LowBit(num) num&(-num)  
  8. #define CLR(arr,val) memset(arr,val,sizeof(arr))  
  9. int total,n,Fx,Fy,Tx,Ty,value,sum,Count=1,C[MAX][MAX];  
  10. void Add(int x,int y,int val)  
  11. {   for(int i=x;i<MAX;i+=LowBit(i))  
  12.         for(int j=y;j<MAX;j+=LowBit(j))  
  13.             C[i][j]+=val;  
  14. }  
  15. int Sum(int x,int y)//求从位置(1,1)到(x,y)的元素之和  
  16. {   int sum=0;  
  17.     for(int i=x;i>0;i-=LowBit(i))  
  18.         for(int j=y;j>0;j-=LowBit(j))  
  19.             sum+=C[i][j];  
  20.     return sum;  
  21. }     
  22. int main()  
  23. {   scanf("%d",&total);  
  24.     while(total--)  
  25.     {   CLR(C,0);  
  26.         for(int i=1;i<MAX;i++)//记得要初始化为1   
  27.             for(int j=1;j<MAX;j++)  
  28.                 Add(i,j,1);  
  29.         printf("Case %d:\n",Count++);  
  30.         scanf("%d",&n);  
  31.         for(int i=0;i<n;i++)  
  32.         {   char c[5];  
  33.             scanf("%s",c);  
  34.             switch(c[0])  
  35.             {   case 'A':  
  36.                     scanf("%d%d%d",&Fx,&Fy,&value);//在(Fx,Fy)处增加value本书   
  37.                     Add(Fx+1,Fy+1,value);  
  38.                     break;  
  39.                 case 'D':  
  40.                     scanf("%d%d%d",&Fx,&Fy,&value);//在(Fx,Fy)处移除value本书,如果这里的书不足,则只移除原有的书  
  41.                     sum=Sum(Fx+1,Fy+1)+Sum(Fx,Fy)-Sum(Fx+1,Fy)-Sum(Fx,Fy+1);//求出(Fx+1,Fy+1)位置的书的数目,可以用容斥定理解决  
  42.                     value=min(value,sum);  
  43.                     Add(Fx+1,Fy+1,-value);   
  44.                     break;  
  45.                 case 'M':  
  46.                     scanf("%d%d%d%d%d",&Fx,&Fy,&Tx,&Ty,&value);//从(Fx,Fy)处移value本书到(Tx,Ty)处,同上面不足的话移除所有的书   
  47.                     sum=Sum(Fx+1,Fy+1)+Sum(Fx,Fy)-Sum(Fx+1,Fy)-Sum(Fx,Fy+1);  
  48.                     value=min(sum,value);  
  49.                     Add(Fx+1,Fy+1,-value);  
  50.                     Add(Tx+1,Ty+1,value);  
  51.                     break;  
  52.                 default:     
  53.                     scanf("%d%d%d%d",&Fx,&Fy,&Tx,&Ty);  
  54.                     if(Fx>Tx) swap(Fx,Tx);  
  55.                     if(Fy>Ty) swap(Fy,Ty);  
  56.                     sum=Sum(Tx+1,Ty+1)+Sum(Fx,Fy)-Sum(Fx,Ty+1)-Sum(Tx+1,Fy);    
  57.                     printf("%d\n",sum);          
  58.             }  
  59.         }  
  60.     }  
  61.     return 0;  
  62. }  

求点(Fx+1,Fy+1)的书的数目：

可知：由面积关系可知，B点的值为：Sum(Fx+1,Fy+1)+Sum(Fx,Fy)-Sum(Fx+1,Fy)-Sum(Fx,Fy+1);

最后的求(Fx,Fy)到(Tx+1,Ty+1)的元素之和同样可求，即求ABCD的面积，只是C(Fx,Fy),B(Tx+1,Ty+1),A(Fx,Ty+1),D(,Tx+1,Fy),这个时候面积为：

Sum(Tx+1,Ty+1)+Sum(Fx,Fy)-Sum(Fx,Ty+1)-Sum(Tx+1,Fy)。

三维树状数组：关键是点的值的变化，如下图，只标了几个点的坐标，O不一定指原点，只是为了立体感强点。

例题4：HDU 3584(Cube)

 

[cpp]  view plain copy print ?
  1. #include<iostream>  
  2. #include<cstring>  
  3. #include<cstdio>  
  4. using namespace std;  
  5. const int MAX=110;  
  6. #define LowBit(num) num&(-num)  
  7. #define CLR(arr,val) memset(arr,val,sizeof(arr))  
  8. int n,m,C[MAX][MAX][MAX];   
  9. void Add(int x,int y,int z,int val)  
  10. {   for(int i=x;i<=n;i+=LowBit(i))  
  11.         for(int j=y;j<=n;j+=LowBit(j))  
  12.             for(int k=z;k<=n;k+=LowBit(k))  
  13.                 C[i][j][k]+=val;   
  14. }  
  15. int Sum(int x,int y,int z)  
  16. {   int sum=0;  
  17.     for(int i=x;i>0;i-=LowBit(i))  
  18.         for(int j=y;j>0;j-=LowBit(j))  
  19.             for(int k=z;k>0;k-=LowBit(k))  
  20.                 sum+=C[i][j][k];  
  21.     return sum;               
  22. }  
  23. int main()  
  24. {   while(scanf("%d%d",&n,&m)!=EOF)  
  25.     {   CLR(C,0);  
  26.         for(int i=0;i<m;i++)  
  27.         {   int comp,Sx,Sy,Sz,Ex,Ey,Ez;  
  28.             scanf("%d",&comp);  
  29.             switch(comp)  
  30.             {   case 0:  
  31.                     scanf("%d%d%d",&Sx,&Sy,&Sz);  
  32.                     printf("%d\n",Sum(Sx,Sy,Sz)&1);  
  33.                     break;  
  34.                 default:  
  35.                     scanf("%d%d%d%d%d%d",&Sx,&Sy,&Sz,&Ex,&Ey,&Ez);  
  36.                     //变成立方体了~~~~   
  37.                     Add(Sx,Sy,Sz,1);    
  38.                     Add(Sx,Ey+1,Ez+1,1);  
  39.                     Add(Ex+1,Sy,Ez+1,1);  
  40.                     Add(Ex+1,Ey+1,Sz,1);  
  41.                     Add(Ex+1,Sy,Sz,1);  
  42.                     Add(Sx,Ey+1,Sz,1);  
  43.                     Add(Sx,Sy,Ez+1,1);  
  44.                     Add(Ex+1,Ey+1,Ez+1,1);     
  45.             }  
  46.         }  
  47.     }  
  48.     return 0;  
  49. }  
涉及到树状数组的题目有： HDOJ 1116，1394，1541，1556，1640，1867，2227，,2275，2430，2492，2642，2668，2689，2838，2852，3047，3450，3743，3887，4031， POJ1195，1990，2029，2155，2299，2352，2464，2481，3067，3321，3468,

 

不可不戒
关注
专栏目录
树状数组（入门附模板）
cyyyyds857的博客
06-21 2063
树状数组或二叉索引树（英语：Binary Indexed Tree），又以其发明者命名为Fenwick树，最早由Peter M. Fenwick于1994年以A New Data Structure for Cumulative Frequency Tables为题发表在SOFTWARE PRACTICE AND EXPERIENCE。其初衷是解决数据压缩里的累积频率（Cumulative Frequency）的计算问题，现多用于高效计算数列的前缀和， 区间和。——源自百度百科。
ACM-数据结构-树状数组I
我了个去
07-13 813
ACM竞赛中，树状数组，即二分索引树(BinaryIndexedTree，BIT)，也是常见的一种数据结构，其应用场景如下： 给出一个长度为n的数组(a[1]-a[n])，每一次给出一个i，询问该数组的前缀和sum[i]。一般情况下，可以在O(n)时间复杂度内处理出所有的前缀和，在O(1)时间复杂度内回答询问。但是如果还存在修改操作，比如修改数组中的某一个元素的值，就无法保证只进行一次O(n)时
树状数组（转载修改添加）
yumenawei的专栏
04-21 396
例题引入：依然以HDU1166为例。问题还是解决查询区间和。上一篇博客的树状数组，虽然简单易懂，但是代码量和空间复杂度明显不够完美。树状数组较好的解决了这两个问题。 一、树状数组是干什么的？ 平常我们会遇到一些对数组进行维护查询的操作，比较常见的如，修改某点的值、求某个区间的和，而这两种恰恰是树状数组的强项！当然，数据规模不大的时候，对于修改某点的值是非常容易的，复杂度是O(1)，但是对
js 树形数据操作
weixin_42630688的博客
05-23 672
树形数据操作 let arr = [ { id: '1', title: '节点1', children: [ { id: '1-1', title: '节点1-1' }, { id: '1-2', title: '节点1-2' } ] },.
js中树形结构修改自定义选择层级属性值
Lingyv77的博客
10-20 2651
js中树形结构修改自定义选择层级属性值
js用闭包遍历树状数组的方法
12-12
做公司项目时，要求写一个方法，方法的参数为一个菜单数组集合和一个菜单id，菜单数组的格式为树状json,如下面所示： 代码如下:[{“id”:28,”text”:”公司信息”,”children”:[  {“id”:1,”text”:”公司文化...
算法NOIP树状数组校门外的树.pdf
05-02
树状数组的核心思想是通过特定的二进制索引操作，将一个一维数组转化为类似树形的数据结构，进而利用这种结构简化更新和查询操作。 在树状数组中，我们不直接维护前缀和数组S[]，而是将其划分为若干个小区间和，并...
树状数组_洛谷_树状数组_
10-02
树状数组，也被称为线段树（Segment Tree），是一种数据结构，主要用于高效地处理动态区间查询和修改问题。在编程竞赛和算法设计中，树状数组是解决许多问题的利器，尤其是在面对需要频繁更新和查询区间信息的问题时...
遍历树，并修改节点字段值
最新发布
qq_39066538的博客
12-27 703
遍历树，并修改节点字段值
JS树形数据递归改造
qq_45232251的博客
11-10 2349
递归改造树形数据
基础dp总结
04-03 1031
dp都忘的差不多了，先回顾一些比较简单的dp。 入门级的dp之数塔。Hdu 2084 点击打开链接 代码：（自顶向下） #include #include #include #include #define CLR(arr,val) memset(arr,val,sizeof(arr)) using namespace std; const int N=104;
前端 遍历树结构数据 方法总结
weixin_43416349的博客
06-11 2505
递归： 方法二： 完善一点的 这里有点怪，他是从数组最后一个元素开始遍历 方法二，正常版
js树结构递归并修改
你离优秀差多少
06-15 927
js树结构递归并修改
js封装更改树形结构指定节点字段的值
想象中
08-15 610
更改树形结构指定节点字段的值
JS树结构数据传输，对应替换属性值
yhh_peter的博客
02-23 889
使用miniui加载树形结构时遇到两个问题： 1、获取的原始树结构的数据字段与需要的字段属性名不一样 2、两个treeGrid表加载同一个树形数据时，第一个加载的表功能会受限（即使使用新的对象直接等于原始数据，两个表依然视作了加载的是同一个对象） 自己整理了一个循环递归遍历，将树结构中的数据字段的属性值一一对应进行转换，用两个对象分别去接取，实现了同一个页面加载相同原始数据，操作分离 function changeInfoDetail(mainData){ var newTree = []; if(!c
Polya定理，Burnside引理
热门推荐
许增强
04-19 1万+
涉及到组合数学的问题，首先是群的概念： 设G是一个集合，*是G上的二元运算，如果(G,*)满足下面的条件： 封闭性：对于任何a,b∈G,有a*b∈G; 结合律：对任何a,b,c∈G有(a*b)*c=a*(b*c); 单位元：存在e∈G,使得对所有的a∈G,都有a*e=e*a=a; 逆元：对于每个元素a∈G,存在x∈G,使得a*x=x*a=e,这个时候记x为a-1，称为a的逆元，那么则称(
K8s 很难么？带你从头到尾捋一遍，不信你学不会
民工哥的博客
02-23 9135
点击关注公众号，回复“1024”获取2TB学习资源！为什么要学习 Kubernetes?虽然 Docker 已经很强大了，但是在实际使用上还是有诸多不便，比如集群管理、资源调度、文件管理等...
10天智能锁项目实战第1天（了解单片机STM32F401RET6和C语言基础）
北豼不太皮的博客
02-21 3661
10天智能锁项目实战第1天（了解单片机STM32F401RET6和C语言基础）一、学习目标二、了解单片机STM32F401RET6三、C语言基础 一、学习目标 二、了解单片机STM32F401RET6 4、STM32F401RE特征 三、C语言基础 1.数据类型 常用2的次方： 2^7 = 128 2^8 = 256 2^15 = 32768 2^16= 65536 51单片机常见的数据类型： char: 占内存1字节 取值范围：-128~127 -2^7 ~ 2
树状数组详解与应用
"这是一份关于树状数组的综合学习资料，主要针对树状数组的基本原理和在POJ编程竞赛中的应用进行了整理。文档中包含了树状数组的核心算法实现，包括`Lowbit`、`Update`和`Getsum`三个关键函数的详细解释。" 树状...
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值