算法复键——树状数组

本文搬运自本人高中时期CSDN博客,若图片加载不出来,可到原文查看:https://blog.csdn.net/zhangtingxiqwq/article/details/162026675

树状数组怎么被我忘光了???

1. 干什么的

简单来说,树状数组支持一下两种操作:

  • 单点加

  • 查询前缀和 \to 区间和查询

如果我们记录的是差分数列,那样子可以也可以实现:

  • 区间加

  • 单点查询

不一定是求和,所有满足交换律的都可以用树状数组实现。比如区间积、区间XOR


2. 怎么实现

现在以单点加、求前缀和为例:

我们定义一种操作 lowbit ,即x二进制下最后一个1

加的时候,对后面所有都有影响,因此是+lowbit

求前缀和就回退就行

1
2
3
4
5
6
7
8
9
10
11
struct TreeG {
int cnt[N], i, ans;
void add(int x, int y) {
while(x <= n) cnt[x] += y, x += x & -x;
}
int qry(int x) {
ans = 0;
while(x) ans += cnt[x], x -= x & -x;
return ans;
}
}s;

注意,树状数组的0位置不可用。如果要用到,请+1。

3. 还能干什么

对于树上问题,如果我们处理的是子树的某些和,同时树的形态不变。那么我们可以转成dfn序后用树状数组来维护。

上面的树也可以是Trie树,于是我们可以拿树状数组维护Trie树。

dp的过程中,可以直接把dp的某一维变成树状数组。

cdq分治的有一维是拿树状数组维护的。