poj1195二维树状数组模板

时间:2021-01-23 04:14:07

二维树状数组和一维的也差不多,改一下add和query函数即可:即按行修改,行内单点修改即可

/*
二维树状数组,询问一个二维区间内的数之和
*/
#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
using namespace std;
#define maxn 1050
int s;
int bit[maxn][maxn];
void add(int x,int y,int a){
for(int i=x;i<=s;i+=i&-i)
for(int j=y;j<=s;j+=j&-j)
bit[i][j]+=a;
}
int query(int x,int y){
int ans=;
for(int i=x;i;i-=i&-i)
for(int j=y;j;j-=j&-j)
ans+=bit[i][j];
return ans;
}
int main(){
int t,q,x,y,sum,l,r,a;
scanf("%d%d",&t,&s);
memset(bit,,sizeof bit);
while(scanf("%d",&q),q<){
if(q==){//区间加上一个数
scanf("%d%d%d",&x,&y,&a);
x++;y++;
add(x,y,a);
}
else if(q==){
scanf("%d%d%d%d",&x,&y,&l,&r);
x++;y++;l++;r++;
printf("%d\n",query(l,r)-query(l,y-)-query(x-,r)+query(x-,y-));
}
}
return ;
}