树状数组学习笔记

Rainbow_Cat_M

已于 2023-07-06 08:57:29 修改

阅读量57

点赞数

分类专栏：数据结构文章标签：算法数据结构

于 2023-07-03 19:36:08 首次发布

本文链接：https://blog.csdn.net/Rainbow_Cat_M/article/details/131520895

版权

数据结构专栏收录该内容

6 篇文章 0 订阅

订阅专栏

树状数组就是将一个大区间分为若干个小区间，小区间的共同特点：

若区间结尾为 $R$ ,则区间长度就等于 $R$ 的“二进制分解”下最小的 $2$ 的次幂，即 $lowbit(R)$ 。如 $lowbit(7)=1 ~ ~ ~7=2^2+2^1+2^0$ 所以 $lowbit(7)=2^0=1$
树状数组就是基于上述思想的数据结构，基本用途是维护序列的前缀和。
我们建立数组 $tr[x]$ 保存序列 $a$ 的区间 $[x-lowbit(x)+1,x]$ 中所有数的和，即 $\sum_{i=x-lowbit(x)+1}^x a[i]$ 。我们可以将 $tr[x]$ 看作树形结构，该结构满足一下性质：
1. 每个内部节点 $tr[x]$ 保存以它为根的子树中所有叶子节点的和。
2. 每个内部节点 $tr[x]$ 的子节点个数等于 $lowbit(x)$ 。
3. 除树根外，每个内部节点 $tr[x]$ 的父节点是 $tr[x+lowbit(x)]$ 。

其中 $lowbit(x)=x ~\&-x$
树状数组支持两个基本操作：

1. 查询前缀和

int sum(int x){
	int res=0;
	for(int i=x;i;i-=lowbit(i))
		res+=tr[i];
	return res;
}

2. 单点增加

void add(int x,int c){
	for(int i=x;i<=n;i+=lowbit(i))
		tr[i]+=c;
}

3. $lowbit$ 操作

int lowbit(int x){
	return x&-x;
}

核心代码：

int lowbit(int x){
	return x&-x;
}
void add(int x,int c){
	for(int i=x;i<=n;i+=lowbit(i))
		tr[i]+=c;
}
int sum(int x){
	int res=0;
	for(int i=x;i;i-=lowbit(i))
		res+=tr[i];
	return res;
}

初始化操作：
1. 直接建立一个全为 0 的数组 $tr$
2. 对每个位置 $x$ 执行 $add(x,a[x])$

y总的图

适用范围：

1. 单点修改，区间查询和

2. 区间修改，单点查询值

3. 区间修改，区间查询和

实质：只能做到单点修改加查询

例子：

1. AcWing 241. 楼兰图腾

以 V 为例：

先正序扫描，利用树状数组求出每一个点的前面大于它的个数

再倒序扫描，利用树状数组求出每一个点的后面大于它的个数

$\Lambda$ 同理

代码：

#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int N=2e5+10;
int tr[N],n,a[N];
ll great[N],low[N];
int lowbit(int x){
	return x&-x;
}
void add(int x,int c){
	for(int i=x;i<=n;i+=lowbit(i))
		tr[i]+=c;
}
int sum(int x){
	int res=0;
	for(int i=x;i;i-=lowbit(i))
		res+=tr[i];
	return res;
}
int main(){
	scanf("%d",&n);
	for(int i=1;i<=n;i++) scanf("%d",&a[i]);
	for(int i=1;i<=n;i++){
		int y=a[i];
		great[i]=sum(n)-sum(y);
		low[i]=sum(y-1);
		add(y,1);
	}
	memset(tr,0,sizeof(tr));
	ll res1=0,res2=0;
	for(int i=n;i>=1;i--){
		int y=a[i];
		res1=res1+(ll)great[i]*(sum(n)-sum(y));
		res2=res2+(ll)low[i]*(sum(y-1));
		add(y,1);
	}
	printf("%lld %lld\n",res1,res2);
	
	return 0;
}

2. AcWing 242. 一个简单的整数问题

区间修改+单点查询

本题指令是“区间增加”+“单点查询”。
利用树状数组巧妙地变为“单点增加”+“区间查询”。
解法：对于指令 $C~l~r~d$ ，我们令 $tr[l]$ 加上 $c$ ， $tr[r+1]$ 减去 $c$ ，这样利用树状数组的性质，这个操作只会影响 $[l,r]$ 区间，其他的地方没有影响。

模板题

代码：

#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int N=1e5+10;
int n,m,a[N];
ll tr[N];
int lowbit(int x){
	return x&-x;
}
void add(int x,int c){
	for(int i=x;i<=n;i+=lowbit(i)) tr[i]+=c;
}
ll sum(int x){
	ll res=0;
	for(int i=x;i;i-=lowbit(i))
		res+=tr[i];
	return res;
}
int main(){
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++){
		scanf("%d",&a[i]);
		add(i,a[i]-a[i-1]);
	}
	while(m--){
		char op[2];
		int l,r,d;
		scanf("%s%d",op,&l);
		if(op[0]=='C'){
			scanf("%d%d",&r,&d);
			add(l,d); add(r+1,-d);
		}
		else{
			printf("%lld\n",sum(l));
		}
	}
	return 0;
}

3. AcWing 243. 一个简单的整数问题2

区间修改+区间查询和

利用两个树状数组

下面是y总的解法

本题指令为“区间增加”+“区间查询”
我们利用树状数组先按照上一题思路对原数组进行差分，得到对应的 $b$
则序列 $a$ 的前缀和 $a[1\sim x]$ 为：
$presum(x)=\sum_{i=1}^x \sum_{j=1}^i b[j]=\sum_{i=1}^x(x-i+1)* b[i]=(x+1)\sum_{i=1}^x b[i]-\sum_{i=1}^x i* b[i]$
所以我们建立两个树状数组 $tr1$ 和 $tr2$ ，对于每条指令 $C~l~r~d$ 执行4个操作:
1. 在树状数组 $tr1$ 中，把位置 $l$ 上的数加 $d$ 。
2. 在树状数组 $tr1$ 中，把位置 $r+1$ 上的数减 $d$ 。
3. 在树状数组 $tr2$ 中，把位置 $l$ 上的数加 $l* d$ 。
4. 在树状数组 $tr2$ 中，把位置 $r+1$ 上的数减 $(r+1)* d$ 。

对于每条指令 $Q~l~r$ ，我们还是拆成 $1\sim r$ 和 $1\sim (l-1)$ 两个部分，二者相减。
$ans=presum(r)-presum(l-1)$

代码：

#include<bits/stdc++.h>
using namespace std;
const int N=1e5+10;
#define ll long long
int n,m,a[N];
ll tr1[N];// b[i]前缀和 
ll tr2[N];// i*b[i]前缀和 
int lowbit(int x){
	return x&-x;
}
void add(ll tr[],int x,ll c){
	for(int i=x;i<=n;i+=lowbit(i)) tr[i]+=c;
}
ll sum(ll tr[],int x){
	ll res=0;
	for(int i=x;i;i-=lowbit(i))
		res+=tr[i];
	return res;
}
ll presum(int x){
	return sum(tr1,x)*(x+1)-sum(tr2,x);
}
int main(){
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++) scanf("%d",&a[i]);
	for(int i=1;i<=n;i++){
		ll b=a[i]-a[i-1];
		add(tr1,i,b);
		add(tr2,i,i*b);
	}
	while(m--){
		char op[2]; int l,r,d;
		scanf("%s%d",op,&l);
		if(op[0]=='C'){
			scanf("%d%d",&r,&d);
			add(tr1,l,d); add(tr2,l,l*d);
			add(tr1,r+1,-d); add(tr2,r+1,(r+1)*-d);
		}
		else{
			scanf("%d",&r);
			printf("%lld\n",presum(r)-presum(l-1));
		}
	}
	return 0;
}

4. AcWing 244. 谜一样的牛

核心算法：二分 $+$ 树状数组

逆序开始，对每一头牛进行二分，直到找到最小的 $sum(x)==h[i]+1$ ， $x$ 即为当前牛的身高

代码：

#include<bits/stdc++.h>
using namespace std;
const int N=1e5+10;
#define ll long long
int n,h[N],ans[N],tr[N];
int lowbit(int x){
	return x&-x;
}
void add(int x,int c){
	for(int i=x;i<=n;i+=lowbit(i))
		tr[i]+=c;
}
int sum(int x){
	int res=0;
	for(int i=x;i;i-=lowbit(i))
		res+=tr[i];
	return res;
}


int main(){
	scanf("%d",&n);
	for(int i=2;i<=n;i++) scanf("%d",&h[i]);
	for(int i=1;i<=n;i++) add(i,1);
	for(int i=n;i>=1;i--){
		int k=h[i]+1;
		int l=1,r=n;
		while(l<r){
			int mid=(l+r)>>1;
			if(sum(mid)>=k) r=mid;
			else l=mid+1;
		}
		ans[i]=r;
		add(r,-1);
	}
	for(int i=1;i<=n;i++) printf("%d\n",ans[i]);
	return 0;
}