树状数组简谈

——2020.8.5
学习笔记思路记录

树状数组：

1.用法
一般用于解决动态前缀和问题，即求数列中某段区间元素的和，过程中元素a[i]的值可以改变，用树状数组可在一定程度上均衡查询与修改操作的时间复杂度，查询/修改的时间复杂度为O(nlogn)
2.lowbit(x)概念引入
获取一个数的2次幂拆分中，最小的2的幂
即二进制中从右往左第一个为1的数及其后面的0
eg 10010 求得 10, 10010001 求得 1
方法:取反->加1->取&
由于取反加一得操作与补码相同，且负数为正数得补码，有
lowbit(x)=x&(-x)
3.结构
抽象图：

第i个位置记录(i-lowbit(i), i]中的数字的和
第i个位置的父亲结点为i+lowbit(i)
4.实现

//代码示例：
int d[500001];//维护的下标从1开始
inline int lowbit(int x){ return x&(-x); }//反回lowbit
void update(int x, int k){  //第x个位置+K 
	while(x<=n){
		d[x]+=k;
		x+=lowbit(x);
	}
}
int getsum(int x){  //求d[1]到d[x]到和
	int ans=0;
	while(x){
		ans+=d[x];
		x-=lowbit(x);
	}
	return ans;
}

求 [ a[i], a[j] ]的所有数得和时直接用 getsum(j)-getsum(i-1)
5.资料
视频（b站上找得一个非常好的视频讲解)
洛谷上的练习题目
P3374 【模板】树状数组 1
P3368 【模板】树状数组 2
两个模板题的题解参考

//洛谷P3374 【模板】树状数组 1
//树状数组
//Author : Sunyi 
//2020.7.27 
#include<iostream>
#include<cstring>
using namespace std;

int n, m;
int d[500001];
inline int lowbit(int x){ return x&(-x); }
void update(int x, int k){  //第x个位置+K 
	while(x<=n){
		d[x]+=k;
		x+=lowbit(x);
	}
}
int getsum(int x){
	int ans=0;
	while(x){
		ans+=d[x];
		x-=lowbit(x);
	}
	return ans;
}

int main()
{
	ios::sync_with_stdio(false);//关闭同步 
	memset(d, 0, sizeof(d));
	cin>>n>>m;
	for(int i=1; i<=n; i++)
	{
		int c;
		cin>>c;
		update(i, c);
	}
	for(int i=1; i<=m; i++)
	{
		int ju, x, y;
		cin>>ju>>x>>y;
		if(ju==1)
			update(x, y);
		else
			cout<<getsum(y)-getsum(x-1)<<"\n";
	}
	return 0;
} 

//P3368 【模板】树状数组 2
//用树状数组维护差分和 
//Author : Sunyi 
//2020.7.27 
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
inline ll read()//快速读入int型
{
	ll x=0,f=1;char ch=getchar();
	while (!isdigit(ch)){if (ch=='-') f=-1;ch=getchar();}
	while (isdigit(ch)){x=x*10+ch-48;ch=getchar();}
	return x*f;
}
ll n, m;
ll d[500001];
inline ll lowbit(ll x){
	return x&(-x);
}
void update(int x, int y){
	while(x<=n){
		d[x]+=y;
		x+=lowbit(x);
	}
}
ll getsum(ll x){
	ll ans=0; 
	while(x!=0){
		ans+=d[x];
		x-=lowbit(x);
	}
	return ans;
}
int main()
{
	memset(d, 0, sizeof(d));
	n=read(), m=read();
	ll last=0;
	for(ll i=1; i<=n; i++)
	{
		ll c=read();
		ll nu=c-last;
		last=c;
		update(i, nu);
	}
	for(ll i=1; i<=m; i++)
	{
		ll ju;
		ju=read();
		if(ju==1)
		{
			ll x, y, k;
			x=read(), y=read(), k=read();
			update(x, k), update(y+1, -k);
		}
		else
		{
			ll h;
			h=read();
			printf("%lld\n", getsum(h)); 
		}
	}
	return 0;
}