数据结构——树状数组

树状数组

**前言：**树状数组现在多用于解决单点修改，求前缀和工作，时间复杂度均为$O(logn)$，空间复杂度为$O(n)$。相比较于线段树，树状数组的作用就显得比较局限，但是它的优点就在于它的常数比较小。线段树在未优化的情况下空间复杂度为$O(2n)$，有时候为了防治溢出往往要开长度为$4n$的数组。

树状数组基本思想

树状数组主要依赖于任何数都可以转化成$2^i$相加的形式，例如$7=2^2+2^1+2^0$，并且如果按照指数从大到小进行区间划分，那么$[1-7]$这个区间可以划分成$[1-4],[5-6],[7,7]$这三个区间。

即有整数$x=2^{i_1}+2^{i_2}+...+2^{i_j}$，可以将区间划分成：$[1-2^{i_1}],[2^{i_1}+1,2^{i_2}]...[2^{i_1}+...2^{i_{j-1}}+1,2^{i_j}]$，共$logn$个区间，并且我们发现每个区间的区间长度恰好为$lowbit(2^i)$

$lowbit(x)$指的是$x$在二进制中，最低位的$1$与它后面的$0$进行的组合。

树状数组的实现

树状数组总共有两个操作，首先是修改操作。定义一个数组$c[x]$，表示从$x-lowbit(x)+1,x$的和。当在第$w$位上加上$num$，那么就要对它后面的数进行更改，它后面的数就是$lowbit(i)$的长度

void add(int w,int num){
	for(int i=w;i<=n;i+=i&(-i)){
		c[i]+=num;
	}
}

查询操作

int query(int x){
	int ans=0;
	for(int i=x;i>=1;i-=i&(-i)){
		ans+=c[i];
	}
	return ans;
}

相关题目

P3374 【模板】树状数组 1

P1908 逆序对

P1972 [SDOI2009] HH的项链