树状数组(BIT)简单分析、应用

YDelete

已于 2024-08-06 00:32:22 修改

阅读量1.9k

点赞数 62

分类专栏：日常分享文章标签：算法 c++ c语言数据结构

于 2024-08-05 20:18:45 首次发布

CC BY-NC-SA 4.0

本文链接：https://blog.csdn.net/Ymy251325/article/details/140935673

版权

日常分享专栏收录该内容

2 篇文章 0 订阅

订阅专栏

树状数组(BIT)简单分析、应用

前言

文章仅作参考、学习

本篇文章作为作者日常分享，相比别的大佬可能质量有点不尽人意，有什么错误请大家指出。

Now

为了解决上次在讲康托展开时没解决的问题——“如何用优化康托展开？”（本来说是用线段树，但线段树过于复杂了，这里再插个锚点🚩，以后再学）；今天我们学习树状数组。（上次关于康托展开（仅展开）的问题（8月6日）已解决，链接可见本文最后）

前置知识

既然学树状数组，那么前缀和、差分应该都学过了吧，这里只简单提一下。先来回顾一些位运算，以及说一下lowbit函数以便后面的树状数组进行节点更新。下面给出一些位运算示例、有关前缀和与差分的知识、lowbit函数的功能和实现。

“&” 与两数都为1时，结果才为1
“~” 取反 0变1，1变0
“^” 异或两个位相同为0，相异为1

F_T1
F_T2

（其中diff表示差分、prefix表示前缀和、arr表示原数组、x表示加的数）

lowbit函数可以用来取一个二进制数最低位的1 以此为最高位往后全为0的数。

比如： $0011_2=3_{10}$ ，使用lowbit函数后可得到 $0001_2=1_{10}$

$00001100_2=12_{10}$ ，使用lowbit函数后可得到 $00000100_2=4_{10}$

lowbit函数实现方式：

return x & (x ^ (x - 1));
return x & -x;//简洁明了，建议使用这个
return x & (~x + 1);//实际上与第二种一样

自己推一下就好了；顺便提醒一下，计算机中数是按补码的形式储存的

树状数组

知识点（规律）

如图所示，这是树状数组的基本结构，tree分成不同的层（由下往上，从1开始）。

观察可得，第i层各数的lowbit都是 $2^{i-1}$ （图示的右边就是各层的lowbit），且对于每个t[j]，它都有i个子节点，子节点由下标从小到大依次是t[j-lowbit(i-1)],t[j-lowbit(i-2)],…,t[j-lowbit(i)]。反过来（属规律，从图上推）也成立，既对于t[i]，它的父节点是t[j+lowbit(j)]。

同时，t[j-lowbit(j)]是t[j]上一层比t[j]下标小的最近的集合，再继续这样往前推，即可得到从a[1]到a[j]的总和，求区间和也一样，再减去区间前面的总和就行了；区间修改直接在原数组的差分数组做BIT上就行了。

感觉上面讲的有点难懂，一大段，没有举例还是太晦涩了，所以下面是主要的概况与举例；

t[i]的父节点是t[i+lowbit(i)]：比如t[7]的父节点是t[7+lowbit(7)]=t[7 + 1]=t[8]；t[2]的父节点是t[2+lowbit(2)]=t[2 + 2]=4[4]
区间a[1, r]的大小为t[r]+t[r-lowbit®]+t[t[r-lowbit[r]-lowbit(r-lowbit[r])]+…(直到当前位减lowbit为0)：比如区间a[1, 5]大小为t[5]+t[5-lowbit(5) = 4];（4 - lowbit(4) = 0）

C++代码

不多说了，直接上代码。

#include <bits/stdc++.h>

using namespace std;

int n, inl, inr, inx;
vector<int> arr;//原数组
vector<int> bit;//树状树状（二元数组）
vector<int> diff;//差分

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

//单点修改
void add(int x,int z){
	int size = bit.size();
	for(int i = z;i <= size;i += lowbit(i)) bit[i] += x;
}

//区间和计算
int summing(int l, int r){
	int sum = 0;
	for(int i = l - 1;i != 0;i -= lowbit(i)) sum -= bit[i];
	for(int i = r;i != 0;i -= lowbit(i)) sum += bit[i];
	return sum;
}

int main(){
	std::ios::sync_with_stdio(false);
	std::cin.tie(nullptr);
	
	cin >> n;//原数组长度
	arr.resize(n + 1, 0);
	bit.resize(n + 1, 0);
	diff.resize(n + 1, 0);
	
	for(int i = 1;i <= n;i++) cin >> arr[i];//原数组
	for(int i = 1;i <= n;i++) add(arr[i], i);//计算前缀和
	for(auto ele : bit) cout << ele << " ";cout << endl;
	
	cin >> inl >> inr;//区间
	cout << summing(inl, inr) << endl;
	bit.clear();
	bit.resize(n + 1, 0);
	
	//差分
	for(int i = 1;i <=n;i++) diff[i] = arr[i] - arr[i - 1];
	for(auto ele : diff) cout << ele << " ";cout << endl;
	for(int i = 1;i <= n;i++) add(diff[i], i);//用树状数组维护差分数组
	cin >> inl >> inr >> inx;//区间、值
	add(inx, inl);
	add(-inx, inr + 1);
	
	//单点查询
	cin >> inr;
	cout << summing(1, inr) << endl;
	
	return 0;
}