重拾树状数组

树状数组

（本文会详细介绍树状数组，萌新也可以轻松看懂）

引言

在做题过程中，我们有时会要维护这样一个前缀和数组：
$Sum[i]=a[1]+a[2]+a[3]+.......+a[i]$
不难发现，如果我们修改了其中一个$a[x]$的值，那么Sum[i]的值会随之改变；
如果我们修改了其中一个$a[i]$的值，那么$Sum[i]+Sum[i+1].....+Sum[n]$的值都会改变。也就是说，每次修改$a[i]$后，前缀和最坏需要$O(n)$的时间调整。当$n$非常大的时候，程序会非常慢。所以我们引入树状数组，它的修改和求和都是$O(nlogn)$级别的。

基本算法

比如一个二进制数：$10101$
其中等于1的位置是$0,2,4$
那么它就可以表示成，$2^0+2^2+2^4$

然后让我们看一下树状数组基本结构：

如图可知：

C[1]=A[1];

C[2]=A[1]+A[2];

C[3]=A[3];

C[4]=A[1]+A[2]+A[3]+A[4];

C[5]=A[5];

C[6]=A[5]+A[6];

C[7]=A[7];

C[8]=A[1]+A[2]+A[3]+A[4]+A[5]+A[6]+A[7]+A[8];

将C[]数组的结点序号转化为二进制

1=(001) C[1]=A[1];

2=(010) C[2]=A[1]+A[2];

3=(011) C[3]=A[3];

4=(100) C[4]=A[1]+A[2]+A[3]+A[4];

5=(101) C[5]=A[5];

6=(110) C[6]=A[5]+A[6];

7=(111) C[7]=A[7];

8=(1000) C[8]=A[1]+A[2]+A[3]+A[4]+A[5]+A[6]+A[7]+A[8];

设节点编号为 x（即结点 C[x]），那么这个节点管辖的区间为 2^k（其中 k 为 x 的二进串末尾 0 的个数）个元素。因为这个区间最后一个元素必然为 A[x]，所以很明显有：
$C[x]=A[x–2^k+1]+\dots +A[x]$
算这个 2^k 有一个快捷的办法——定义一个如下的 lowbit 函数即可

lowbit函数

int lowbit(x)//把x二进串的最低位（最右边）1取出 
{
	return x&(-x);
} 

例如6，二进制为110，最右边的1是第二位，也可以理解成是最低位1和它后边的0构成的数值，6就是10，十进制的2。

一些操作

1.对某个元素进行加法操作
树状数组支持单点修改，即给序列中的一个a[x]+y，同时正确维护序列的前缀和。任意一个节点的祖先至多也只有logn个，逐一修改即可。

void update(int x,int y)
{
	while(x<=n)
	{
		c[x]+=y;
		x+=lowbit(x);
	}
 } 

2.查询前缀和

//step1: 令 sum = 0，转第二步；
//step2: 假如 n <= 0，算法结束，返回 sum 值，否则 sum = sum + Cn，转第三步；
//step3: 令 n = n-lowbit(n)，转第二步。
int sum(int x)
{
	int s=0;
	while(x>0)
	{
		s+=c[x];
		x-=lowbit(x);
	}
	return s;
}

其他

树状数组的效率都是logn级别的：
n=n–lowbit(n) 这一步实际上等价于将 n 的二进制的最后一个 1 减去。而 n 的二进制里最多有 log(n)个 1，所以查询效率是 log(n)的。那么修改的效率呢? 修改一个节点，必须修改其所有的祖先，最坏情况下为修改第一个元素，最多有 log(n) 个祖先。
还有下标不能为0，lowbit(0)=0，会陷入死循环。

打板子

P3374 【模板】树状数组 1

//P3374
#include <bits/stdc++.h>
using namespace std;

int n,m;
int c[500005];

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

int sum(int x)
{
	int s=0;
	while(x>0)
	{
		s+=c[x];
		x-=lowbit(x);
	}
	return s;
}

void update(int x,int y)
{
	while(x<=n)
	{
		c[x]+=y;
		x+=lowbit(x);
	}
}

int main()
{
	ios::sync_with_stdio(0);
	cin>>n>>m;
	for(int i=1;i<=n;i++)
	{
		int a;
		cin>>a;
		update(i,a);
	}
	while(m--)
	{
		int op,x,y;
		cin>>op>>x>>y;
		if(op==1) update(x,y);
		else cout<<sum(y)-sum(x-1)<<endl;
	}
	return 0;
 } 

To Be Continued