概念
什么是树状数组???
就是把一个普通数组转化为树的形式,那就变成了树状数组。
先上张图。
树状数组d[i]存储的就是前lowbit(i)的a[i-lowbit(i)+1]的总和。
性质
1.i节点子节点个数为lowbit(i)
2.i节点的父节点为i+lowbit(i)
3.树的深度是(logN)
实现
n=read();
for(int i=1;i<=n;++i)
{
a[i]=read();
int ii=i;
while (ii<=n)
{
d[ii]+=a[ii];
ii+=lowbit(ii);
}
}
那么转化为树状数组有什么用呢???
那么下面讲讲树状数组的两个操作。
操作
修改单点
就是修改原数组,导致的树状数组的修改。复杂度是O(logN)。
原数组修改,那么i的各父节点也需要进行修改。。
想想树向上返回的过程,也就是修改的过程。
代码与上相似。
while (ii<=n)
{
d[ii]+=xx-a[ii];//xx为要修改的值
a[ii]=xx;
ii+=lowbit(ii);
}
这些修改操作其实都是为区间和查询服务的。
区间和查询
由于每个节点都是求了前lowbit(i)这个区间的和。
求区间【L,R】就只要求【1,R】-【1,L-1】
其复杂度为O(logN)
也就是ask(R)-ask(L-1)
int ask(int x)
{
int ans=0;
while (x)
{
ans+=d[x];
x-=lowbit(x);
}
}