一.定义:
是一个查询和修改复杂度都为log(n)的数据结构。
主要用于查询任意两位之间的所有元素之和,但是每次只能修改一个元素的值;经过简单修改可以在log(n)的复杂度下进行范围修改,但是这时只能查询其中一个元素的值(如果加入多个辅助数组则可以实现区间修改与区间查询)。
二.功能:
区间和查询、单点修改
三.与线段树的比较:
能用树状数组解决的问题,基本上都能用线段树解决,而线段树能解决的树状数组不一定能解决。树状数组代码短,时空常数小。但相比较而言,树状数组效率要高很多。
四.构造树状数组(以区间和为例):
构造原理:
仅将必要数组元素,存入树状数组中。如果可通过其中已存入的数据推算,则不存入。如c[2]=A[1,2]-A[1],c[3,4]=A[1,4]-A[1,2]…所以c[2],c[3,4]…不存入。
特点:
奇数结点:A[1]=[1]=c[1],A[3]==[3]=c[3]…
偶数结点:A[2]=[1,2]=c[1]+c[2],A[4]=[1,4]=A[2]+c[3]+c[4]…
1.结点编号:x
2.管辖区间元素个数:2^k。(k为二进制最后一个1的后面的0的个数)
eg:
2:10 管辖区间个数:1。管辖区间元素个数:2^1=2。
6:10 管辖区间个数:1。管辖区间元素个数:2^1=2。
8:1000 管辖区间个数:3。管辖区间元素个数:2^3=8。
3.利用函数求k:
int lowbit(int x)
{
return x&(-x);
}
五.代码(同样以区间和为例)
1.构建树状数组
//x输入元素在数组中的位置,z该元素的值,n该数组的长度
void build(int x,int z,int n)
{
while(x<=n)
{
t[x]+=z;
x+=lowbit(x);
}
return;
}
2.前x个数的和
int sum(int x)
{
int ans=0;
while(x>0)
{
ans+=t[x];
x-=lowbit(x);
}
return ans;
}
3.求区间和
int qjh(int l,int r)
{
return sum(r)-sum(l-1);
}
4.主函数
int main()
{
cin>>n>>m;
for(i=1;i<=n;i++)
{
cin>>a[i];
bulid(i,a[i]);
}
for(i=1;i<=m;i++)
{
cin>>l>>r;
cout<<qjh(l,r)<<endl;
}
return 0;
}
注:来源为百度和牛客算法竞赛入门班笔记
此外如有错误,欢迎指出