树状数组(Binary Indexed Tree,简称BIT)是一种用于高效处理数组范围查询和单点更新操作的数据结构。它解决了传统的线段树在处理范围查询时的低效问题,并且在某些情况下,比如查询和更新操作频繁的情况下,它比线段树更快。
树状数组的核心思想是将数组的每个位置i
表示为一个二进制数,然后用这个二进制数的最后一个1来确定它是哪个父节点的子节点。更具体地说,如果i
的二进制表示中最后一个1位于第k
位(从右往左数,起始位为0),那么i
就属于它的父节点。
下面是树状数组的C++实现:
#include <iostream>
#include <vector>
// 树状数组类
class BinaryIndexedTree {
private:
int n; // 数组大小
std::vector<int> tree; // 树状数组
// lowbit函数,获取x的最后一个1的位置
int lowbit(int x) {
return x & -x;
}
public:
// 构造函数,初始化树状数组
BinaryIndexedTree(int size) : n(size), tree(size + 1, 0) {}
// 单点更新,将index位置的值增加delta
void update(int index, int delta) {
index++; // 树状数组的索引从1开始
while (index <= n) {
tree[index] += delta;
index += lowbit(index);
}
}
// 查询前缀和,即[1, index]的和
int query(int index) {
int sum = 0;
index++; // 树状数组的索引从1开始
while (index > 0) {
sum += tree[index];
index -= lowbit(index);
}
return sum;
}
// 查询区间和,即[left, right]的和
int queryRange(int left, int right) {
return query(right) - query(left - 1);
}
};
int main() {
// 测试代码
BinaryIndexedTree bit(10); // 创建一个大小为10的树状数组
bit.update(0, 1); // 将第1个元素更新为1
bit.update(1, 2); // 将第2个元素更新为2
bit.update(2, 3); // 将第3个元素更新为3
std::cout << "Sum of elements from 1 to 3: " << bit.queryRange(1, 3) << std::endl; // 输出1到3的和
return 0;
}
在这个实现中,我们使用一个数组来存储树状数组的节点值。
函数用于单点更新,它通过循环加上
函数计算出的差值来更新所有受影响的节点。
函数用于查询前缀和,它通过循环减去
函数计算出的差值来累加所有相关的节点值。
函数用于查询区间和,它通过两个
调用来计算出区间和。
请注意,树状数组的索引是从1开始的,所以我们在所有操作中都对索引进行了加1的处理。此外,树状数组的大小通常是原始数组的大小加上1,以确保我们可以安全地使用索引来访问所有的节点。