一维和二维树状数组的使用
1.一维树状数组的使用
c[i]:以i结尾的树状数组规律计数和
注:下标均从1开始
int low_bit(int x){//k是x的二进制中从右往左数0的个数,返回2^k
return x&(-x);
}
int Sum(int n){//求a[1...n]的和
int sum=0;
while(n){
sum+=c[n];
n-=low_bit(n);
}
return sum;
}
void update(int i,int x){//更新a[i]=x
while(i<=n){
c[i]+=x;
i+=low_bit(i);
}
}
2.二维树状数组 修改某个点的权值;统计矩形x1,y1,x2,y2内的点权值之和
c[i][j]表示以坐标(i,j)结尾的树状数组规律计数和
#include<stdio.h>
#include<iostream>
#include<string.h>
#define maxn 1030
#define mem(a,b) memset(a,b,sizeof(a))
using namespace std;
int n,q,ans;
int c[maxn][maxn];
inline int lowbit(int x){
return ((x)&(-x));
}
void update(int x,int y,int k){
for(int i=x;i<=n;i+=lowbit(i))
for(int j=y;j<=n;j+=lowbit(j))
c[i][j]+=k;
}
int query(int x,int y){//求矩形1,1,x,y内的权值之和(两边均为闭区间)
int sum=0;
for(int i=x;i>0;i-=lowbit(i))
for(int j=y;j>0;j-=lowbit(j))
sum+=c[i][j];
return sum;
}
int main(){
scanf("%d",&n);
while(scanf("%d",&q)!=EOF){
if(q==3) break;
if(q==1){
int x,y,k;
scanf("%d%d%d",&x,&y,&k);
update(x+1,y+1,k);
}
if(q==2){
int x1,y1,x2,y2;
scanf("%d%d%d%d",&x1,&y1,&x2,&y2);
ans=query(x2+1,y2+1)-query(x1,y2+1)-query(x2+1,y1)+query(x1,y1);
printf("%d\n",ans);
}
}
return 0;
}