数据结构笔记-树状数组

数据结构笔记-树状数组

能解决的问题：
（1）快速求前缀和 O(nlogn)
（2）修改某一个数 O(nlogn)

对比于前缀数组和差分数组

1.基本原理

（1）定义：

树状数组（Binary Indexed Tree, Fenwick Tree），是一种用于高效处理对一个存储数字的列表进行更新及求前缀和、区间和的数据结构。
什么是前缀和以及区间和？前缀和是一个数组的某项下标之前(包括此项元素)的所有数组元素的和。 区间和就是给定区间的数组的和。
它的查询和修改的时间复杂度都是log(n)，空间复杂度则为O(n)。这正是使用树状数组的原因。
（碰到更新元素、求前缀和、区间和就可以考虑树状数组）。

一段长度为x的线段可以被k个2ⁿ线段表示：
即：

x = 2ⁱ¹ + 2ⁱ² + 2ⁱ³ + 2ⁱ⁴ + 2ⁱ⁵ + 2ⁱ⁶ + … 2^ik

也就是长度为x的序列，能够被划分成长度不一的区间。而树状数组储存的这些长度不一的区间和。
（类比与前缀和数组res[i] 储存的是 [0,i]的区间和）

代码：

int lowbit(int x){
    return x&(-x);
}

lowbit：是求一个数字的二进制表示的1所在的最低位（k），而lowbit=2^k。

void add(int x, int v)
{
    for (int i = x; i < N; i += lowbit(i)) tr[i] += v;
}

在x的位置上加上v

int query(int x)
{
    int res = 0;
    for (int i = x; i; i -= lowbit(i)) res += tr[i];
    return res;
}

查询，小于等于x的个数有多少个

2.简单扩展

（1）.+差分

（2）.+ 差分 + 推公式