树状数组是一个查询和修改复杂度都为log(n)的数据结构。主要用于数组的单点修改&&区间求和.正所谓用树状数组能写的题,用线段树都能写,但用线段树能写的用树状数组不一定能写。但奈何树状数组写着简单啊
比如说8这个节点,里面的值是原始数组中[5,8]的和
lowbit:求x最低位的一
int lowbit(x){return x&(-x);}
单点更新:
void update(int x,int y,int n){
for(int i=x;i<=n;i+=lowbit(i)) //x为更新的位置,y为更新后的数,n为数组最大值
c[i] += y;
}
单点查询:
int getsum(int x){
int ans = 0;
for(int i=x;i;i-=lowbit(i))
ans += c[i];
return ans;
}
校门外的树:
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
int n,m,k,l,r;
int a[100010],b[100010];
int lowbit(int x){
return x&(-x);
}
void add1(int x,int y){
while(x<=n){
a[x]+=y;
x+=lowbit(x);
}
}
void add2(int x,int y){
while(x<=n){
b[x]+=y;
x+=lowbit(x);
}
}
int sum1(int x){
int ans=0;
while(x>0){
ans+=a[x];
x-=lowbit(x);
}
return ans;
}
int sum2(int x){
int ans=0;
while(x>0){
ans+=b[x];
x-=lowbit(x);
}
return ans;
}
int main(){
scanf("%d %d",&n,&m);
for(int i=1;i<=m;i++){
scanf("%d %d %d",&k,&l,&r);
if(k==1){
add1(l,1);
add2(r,1);
}
else
printf("%d\n",sum1(r)-sum2(l-1));
}
return 0;
}
区间修改,单点查询
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int maxn=1e6+10;
int n,q,x;
long long a[maxn];
int lowbit(int x){
return x&(-x);
}
void add(int y,int z){
while(y<=n){
a[y]+=z;
y+=lowbit(y);
}
}
long long find(int y){
long long ans=0;
while(y>0){
ans+=a[y];
y-=lowbit(y);
}
return ans;
}
int main()
{
scanf("%d %d",&n,&q);
int qq=0;
for(int i=1;i<=n;i++){
scanf("%d",&x);
add(i,x-qq);//
qq=x;//
}
for(int i=1;i<=q;i++){
scanf("%d",&x);
if(x==1){
int l,r,z;
scanf("%d %d %d",&l,&r,&z);
add(l,z);
add(r+1,-z);
}
else
{
scanf("%d",&x);
printf("%lld\n",find(x));
}
}
return 0;
}