- 与线段树复杂度同级,但是树状数组的常数明显优于线段树,其编程复杂度也远小于线段树。
- 树状数组的作用被线段树完全涵盖,凡是可以使用树状数组解决的问题,使用线段树一定可以解决,但是线段树能解决的问题树状数组未必能解决。
- 树状数组的突出特点是其编程的极端简洁性,使用 lowbit 技术可以在很短的几步操作中完成树状数组的核心操作,其代码效率远高于线段树。
lowbit() 运算:非负整数 x 在二进制表示下最低位1及其后面的0构成的数值。
ex:lowbit(
[
12
]
10
[12]_{10}
[12]10) = lowbit(
[
1100
]
2
[1100]_2
[1100]2) =
[
100
]
2
[100]_2
[100]2 =
[
4
]
10
[4]_{10}
[4]10
int lowbit(int x)
{
return x & -x;
}
树状数组的本质思想是使用树结构维护”前缀和”,从而把时间复杂度降为O(logn)。
- 每个结点t[x]保存以x为根的子树中叶结点值的和
- 每个结点覆盖的长度为lowbit(x)
- t[x]结点的父结点为t[x + lowbit(x)]
- 树的深度为 l o g 2 n + 1 log_2n+1 log2n+1
单点更新
add(x, k)表示将序列中第x个数加上k。
以add(3, 5)为例:
在整棵树上维护这个值,需要一层一层向上找到父结点,并将这些结点上的t[x]值都加上k,这样保证计算区间和时的结果正确。时间复杂度为O(logn)。
void add(int x, int k)
{
for(int i = x; i <= n; i += lowbit(i))
t[i] += k;
}
区间查询
ask(x)表示将查询序列前x个数的和
以ask(7)为例:
查询这个点的前缀和,需要从这个点向左上找到上一个结点,将加上其结点的值。向左上找到上一个结点,只需要将下标 x -= lowbit(x),例如 7 - lowbit(7) = 6。
int ask(int x)
{
int sum = 0;
for(int i = x; i; i -= lowbit(i))
sum += t[i];
return sum;
}