树状数组(洛克模板)

树状数组适用于O(nlogn)来计算前缀和

学习路径:http://t.csdnimg.cn/ACjd9

例如:

题目1:单点修改,区间查询。

利用普通方法只能部分过数据:

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
ll n;
ll a[1000010]={0},sum[1000010]={0};
//普通方法+前缀和 
int main(){
	ll n,m,l,x,k;
	cin>>n>>m;
	for(int i=1;i<=n;i++) 
	{
		cin>>a[i];
		sum[i]=sum[i-1]+a[i];
	}
	while(m--)
	{
		cin>>l>>x>>k;
		if(l==1)
		{
			a[x]+=k;
			for(int i=x;i<=n;i++)
			{
				sum[i]+=k;
			}
		}
		else
		{
			cout<<sum[k]-sum[x-1]<<endl;
		}
	}
} 

利用树状数组对其进行优化:

树状数组三个基本操作:

1. 更新数据。例如数组中某个元素增加多少,利用Upadate方法对树状数组t[]进行更新;其次在输入原始数组时,利用更新操作建立起树状数组。

2. 求[L,R]的元素之和。利用Sum方法。

3. lowbit方法。利用该方法用于计算x&-x,方便利用加减lowbit访问树状数组的下标。

ll t[1000010]={0};//树状数组 
//树状数组+前缀和
ll lowbit(ll x)
{
	return x&(-x);
}
//更新数据
void Update(ll x,ll a) 
{
	for(ll i=x;i<=n;i+=lowbit(i))
	{
		t[i]+=a;
	}
}
//区间和
ll Sum(ll x,ll y)
{
	ll sum=0; 
	for(ll i=x-1;i;i-=lowbit(i))//计算[1,L-1]的各元素之和
	{
		sum-=t[i];
	}
	for(ll i=y;i;i-=lowbit(i))//计算[1,R]的各元素之和
	{
		sum+=t[i];
	}
	return sum;//返回[L,R]该区间的元素和
}
int main(){
	ll m,l,x,k,i;
	cin>>n>>m;
	for(i=1;i<=n;i++) 
	{
		cin>>a[i];
	    Update(i,a[i]);//在输入数组的时候形成树状数组 
	}
	while(m--)
	{
		cin>>l>>x>>k;
		if(l==1)Update(x,k);//单点修改
		else cout<<Sum(x,k)<<endl;//区间查询
	}
	return 0;
}

题目2:区间修改,单点查询

题目:

差分数组性质:
1. a[x]=d[1]+....d[x]; 计算每个数字实际上是在求d[]的[1,x]的前缀和 。

2. 让区间[L,R]都增加k,我们知道a[i]=a[i-1]+d[i],实际只要使d[L]+k;d[R+1]-k;就可以使得原来数组的该区间每个元素的数值都增加k。

利用以上性质可知,修改区间的所有值只需要对差分数组进行处理,而某个值则是对差分数组[1,x]求和,则可以利用树状数组进行优化时间复杂度。

 普通方法:

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
ll n;
ll a[1000010]={0};
ll d[1000010]={0};//差分数组 的树状数组  
int main()
{
	ll m,i,l,x,k,y,b=0;
	cin>>n>>m;
	for(i=1;i<=n;i++)
	{
		cin>>a[i];
		d[i]=a[i]-a[i-1]; 
	}
	while(m--)
	{
		cin>>l;
		if(l==1)
		{
			cin>>x>>y>>k;
			d[x]+=k;
			d[y+1]-=k;//利用差分数组性质
		}
		else
		{
			cin>>x;
			b=0;
			//for(ll j=1;j<=n;j++) cout<<d[j]<<" ";
			for(ll j=1;j<=x;j++) b+=d[j]; //利用差分数组性质
			cout<<b<<endl;
		}
	} 
	
	return 0;
} 

利用树状数组:
 

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
ll n;
ll a[1000010]={0};
ll d[1000010]={0};//差分数组 的树状数组 

ll lowbit(ll x)
{
	return x&-x;
}
void Update(ll x,ll a)
{
	for(ll i=x;i<=n;i+=lowbit(i))
	{
		d[i]+=a;
	}
}
ll Sum(ll x)//求某个位置x的值
{
	ll sum=0;
	for(ll i=x;i;i-=lowbit(i))
	{
		sum+=d[i];
	}
	return sum;
}
int main()
{
	ll m,i,l,x,k,y,b=0;
	cin>>n>>m;
	for(i=1;i<=n;i++)
	{
		cin>>a[i];
		ll b=a[i]-a[i-1];
	    Update(i,b);//用差分数组形成树状数组
	}
	while(m--)
	{
		cin>>l;
		if(l==1)
		{
			cin>>x>>y>>k;
			Update(x,k);//更新树状数组的和
			Update(y+1,-k);
		}
		else
		{
			cin>>x;
			cout<<Sum(x)<<endl;
		}
    }
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值