树状数组
树状数组利用二进制在线性结构上模拟了树形结构,解决的问题比较集中,但的确是一种极其巧妙的数据结构。
树状数组以二进制为基础,在数组中建立了树形关系,通过加减lowbit(x)实现树上路径的模拟,比较基础的一类问题是树状数组单点修改,此时树状数组可以达到O(logn)。
补充lowbit(x):lowbit(x)表示x的二进制上各位中为“1”的最低位,例如6(110)的lowbit(6)即2(10)。
数组c存储树状数组,树状数组求区间和类似前缀和,从第一位开始将本单元(第i位)数据累加到i+lowbit(i)单元上,修改时仅需按这样的路径依次修改父节点知道根节点,访问前n位和时从第n位开始依次访问n-lowbit(n)单元,这是两条不同的路径,这样操作可以保证前缀和覆盖全部单元。
以下为基础的树状数组单点修改和区间和查询代码
#include<cstdio>
#include<iostream>
using namespace std;
int a[501000],s[501000],n,m,k,l,r;
int lowbit(int x)
{
return x&(-x);
}
void add(int x,int k)
{
while(x<=n)
{
s[x]+=k;
x=x+lowbit(x);
}
}
int find(int x)
{
int num=0;
while(x>0)
{
num+=s[x];
x=x-lowbit(x);
}
return num;
}
int main()
{
cin>>n>>m;
for(int i=1;i<=n;i++)
{
scanf("%d",&a[i]);
add(i,a[i]);
}
for(int i=1;i<=m;i++)
{
scanf("%d%d%d",&k,&l,&r);
if(k==1)
{
add(l,r);
}
else
{
int sum1=find(l-1);
int sum2=find(r);
printf("%d\n",sum2-sum1);
}
}
return 0;
}
区间修改,单点查询略。
区间修改,区间查询可参考https://www.cnblogs.com/lcf-2000/p/5866170.html