树状数组的模板

最新推荐文章于 2025-05-25 19:04:27 发布

阿磊AC

最新推荐文章于 2025-05-25 19:04:27 发布

阅读量473

点赞数

CC 4.0 BY-SA版权

分类专栏：数据结构

本文链接：https://blog.youkuaiyun.com/Joe19310/article/details/77017244

数据结构专栏收录该内容

21 篇文章

订阅专栏

本文介绍了树状数组的基本概念，包括一维和二维树状数组的实现原理与操作方法。通过具体的代码示例，展示了如何使用树状数组进行区间求和等操作。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

//树状数组的模板；
int lowbit(int i){
	return i&(-i);
}//lowbit(i)=2^k(其中k为i在二进制下末尾0的个数)

//下面的代码给原数组的下标为pos的位置上的元素a[pos]加上一个数num;

void  update(int pos,int num){
	while(pos<=n){//n为元素的个数 
		c[pos]=c[pos]+num;
		pos+=lowbit(pos);
	}
} 

//计算原数组A[1]到a[x]的和；
int sum(int x){
	int sum=0;
	while(x>0){
		sum+=c[x];
		x-=lowbit[x];
	}
	return sum;
} 
//二维树状数组
//作用：用于快速求数字的子矩阵的和；
void uodate (int i,int j,int k){//给a[i][j]加上k 
     while(i<=n){
     	int temp=j;
     	while(temp<=n){
     		c[i][temp]+=k;
     		temp+=lowbit(temp);
		 }
		 i+=lowbit(i);
	 }
} 
int sum(int i,int j){//查询a[1][1]到a[i][j]的和
       int sum=0;
	   while(i>0){
	   	 int temp=j;
	   	 while(temp>0){
	   	         sum+=c[i][temp];
	   	         temp-=lowbit(temp);
			}
			i-=lowbit(i);
	   } 
	   return sum;
}