树状数组
(本文会详细介绍树状数组,萌新也可以轻松看懂)
引言
在做题过程中,我们有时会要维护这样一个前缀和数组:
S
u
m
[
i
]
=
a
[
1
]
+
a
[
2
]
+
a
[
3
]
+
.
.
.
.
.
.
.
+
a
[
i
]
Sum[i]=a[1]+a[2]+a[3]+.......+a[i]
Sum[i]=a[1]+a[2]+a[3]+.......+a[i]
不难发现,如果我们修改了其中一个
a
[
x
]
a[x]
a[x]的值,那么Sum[i]的值会随之改变;
如果我们修改了其中一个
a
[
i
]
a[i]
a[i]的值,那么
S
u
m
[
i
]
+
S
u
m
[
i
+
1
]
.
.
.
.
.
+
S
u
m
[
n
]
Sum[i]+Sum[i+1].....+Sum[n]
Sum[i]+Sum[i+1].....+Sum[n]的值都会改变。也就是说,每次修改
a
[
i
]
a[i]
a[i]后,前缀和最坏需要
O
(
n
)
O(n)
O(n)的时间调整。当
n
n
n非常大的时候,程序会非常慢。所以我们引入树状数组,它的修改和求和都是
O
(
n
l
o
g
n
)
O(nlogn)
O(nlogn)级别的。
基本算法
比如一个二进制数:
10101
10101
10101
其中等于1的位置是
0
,
2
,
4
0,2,4
0,2,4
那么它就可以表示成,
2
0
+
2
2
+
2
4
2^0+2^2+2^4
20+22+24
然后让我们看一下树状数组基本结构:
如图可知:
C[1]=A[1];
C[2]=A[1]+A[2];
C[3]=A[3];
C[4]=A[1]+A[2]+A[3]+A[4];
C[5]=A[5];
C[6]=A[5]+A[6];
C[7]=A[7];
C[8]=A[1]+A[2]+A[3]+A[4]+A[5]+A[6]+A[7]+A[8];
将C[]数组的结点序号转化为二进制
1=(001) C[1]=A[1];
2=(010) C[2]=A[1]+A[2];
3=(011) C[3]=A[3];
4=(100) C[4]=A[1]+A[2]+A[3]+A[4];
5=(101) C[5]=A[5];
6=(110) C[6]=A[5]+A[6];
7=(111) C[7]=A[7];
8=(1000) C[8]=A[1]+A[2]+A[3]+A[4]+A[5]+A[6]+A[7]+A[8];
设节点编号为 x(即结点 C[x]),那么这个节点管辖的区间为 2^k(其中 k 为 x 的二进串末尾 0 的个数)个元素。因为这个区间最后一个元素必然为 A[x],所以很明显有:
C
[
x
]
=
A
[
x
–
2
k
+
1
]
+
…
+
A
[
x
]
C[x] = A[x – 2^k + 1]+ … + A[x]
C[x]=A[x–2k+1]+…+A[x]
算这个 2^k 有一个快捷的办法——定义一个如下的 lowbit 函数即可
lowbit函数
int lowbit(x)//把x二进串的最低位(最右边)1取出
{
return x&(-x);
}
例如6,二进制为110,最右边的1是第二位,也可以理解成是最低位1和它后边的0构成的数值,6就是10,十进制的2。
一些操作
1.对某个元素进行加法操作
树状数组支持单点修改,即给序列中的一个a[x]+y,同时正确维护序列的前缀和。任意一个节点的祖先至多也只有logn个,逐一修改即可。
void update(int x,int y)
{
while(x<=n)
{
c[x]+=y;
x+=lowbit(x);
}
}
2.查询前缀和
//step1: 令 sum = 0,转第二步;
//step2: 假如 n <= 0,算法结束,返回 sum 值,否则 sum = sum + Cn,转第三步;
//step3: 令 n = n-lowbit(n),转第二步。
int sum(int x)
{
int s=0;
while(x>0)
{
s+=c[x];
x-=lowbit(x);
}
return s;
}
其他
树状数组的效率都是logn级别的:
n=n–lowbit(n) 这一步实际上等价于将 n 的二进制的最后一个 1 减去。而 n 的二进制里最多有 log(n)个 1,所以查询效率是 log(n)的。那么修改的效率呢? 修改一个节点,必须修改其所有的祖先,最坏情况下为修改第一个元素,最多有 log(n) 个祖先。
还有下标不能为0,lowbit(0)=0,会陷入死循环。
打板子
//P3374
#include <bits/stdc++.h>
using namespace std;
int n,m;
int c[500005];
int lowbit(int x)
{
return x&(-x);
}
int sum(int x)
{
int s=0;
while(x>0)
{
s+=c[x];
x-=lowbit(x);
}
return s;
}
void update(int x,int y)
{
while(x<=n)
{
c[x]+=y;
x+=lowbit(x);
}
}
int main()
{
ios::sync_with_stdio(0);
cin>>n>>m;
for(int i=1;i<=n;i++)
{
int a;
cin>>a;
update(i,a);
}
while(m--)
{
int op,x,y;
cin>>op>>x>>y;
if(op==1) update(x,y);
else cout<<sum(y)-sum(x-1)<<endl;
}
return 0;
}
To Be Continued