树状数组适用于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;
}
}
}