【转载】树状数组（简单）

最新推荐文章于 2021-12-23 11:07:58 发布

Dream_Lolita

最新推荐文章于 2021-12-23 11:07:58 发布

阅读量189

点赞数

分类专栏：学习知识up

本文链接：https://blog.csdn.net/Dream_Lolita/article/details/52149003

版权

学习知识up 专栏收录该内容

28 篇文章 0 订阅

订阅专栏

树状数组(Binary Indexed Tree(BIT), Fenwick Tree)是一个查询和修改复杂度都为log(n)的数据结构。主要用于查询任意两位之间的所有元素之和，但是每次只能修改一个元素的值

#include<cmath>
#include<iostream>
#define MAXN 100000
using namespace std;

int Lowbit(int);
void Change(int,int);
int Query(int);
int Getindex(int);

int n,m,s,x1,x2;
int a[MAXN];
char ch;

int main()
{
	cin>>n>>m;
	for (int i=1;i<=n;i++) 
	{
		cin>>s;
		Change(i,s);
	}
	for (int i=0;i<m;i++)
	{
		cin>>ch;
		switch (ch)
		{
			case 'C':
			{
				cin>>x1>>x2;
				Change(x1,x2);
				break;
			}
			case 'Q':
			{
				cin>>x1>>x2;
				cout<<Query(x2)-Query(x1-1)<<endl;	
				break;
			}
		}
	}
}

int Lowbit(int n)
{
	return (n&-n);
}

void Change(int k,int n)
{
	while (k<=MAXN)
	{
		a[k]+=n;
		k+=Lowbit(k);
	}
}

int Query(int k)
{
	int ans=0;
	while (k>0)
	{
		ans+=a[k];
		k-=Lowbit(k);
	}
	return ans;
}

这是一个最简洁的树状数组。只能用最简单的修改和查询区间和的功能。

树状数组的重要技术：Lowbit（低位技术）

在计算机中，负数以其正值的补码形式表达，方法为其绝对值求反加一。

　　例如 -100的绝对值为100
　　100原码：1000 0000 0110 0100
　　100反码：1111 1111 1001 1011
　　100补码：1111 1111 1001 1100=1111 1111 1001 1011（反码） +1
　　所以-100二进制表示为：1111 1111 1001 1100
　　原码：一个整数按绝对值大小转换成的二进制数称为原码；
　　反码：将二进制数按位取反，所得的新二进制数称为原二进制数的反码；（取反操作：1变0,0变1）
　　补码：反码加1称为补码。

所以 x&(-x) 会返回x二进制之中最后的1和后面的0

如:

x =1： 1 &－１（设位数为８）0000 0001 & 1111 1111 = 1

x = 6:6 & -6 0000 0110&1111 1010 = 2

通过Lowbit，我们就能很快的从查询树或更新树的叶节点一路找到自己的父亲，从而达到用O(log2(n))的时间复杂度来完成查询区间和的目的

树状数组与线段树的比较：

1、树状数组省空间，树状数组规模为n左右，线段树为2n，在加上线段树需要维护其他数据域，大约花费比树状数组大 10倍左右

2、易于编码，树状数组的编码复杂的较为简单

这些都是树状数组的基本操作，需要熟练掌握