HDU 1892 二维树状数组 模板题

假设每个格子代表一个数A[i][j]，i是横坐标，j是纵坐标，左上角的坐标为（1，1)我们要求红色区域元素之和，设sum(i,j)表示以i,j坐标为右下角坐标，以0,0为左上角坐标矩形内的元素之和，

sum(c,d):绿色+黄色+红色+蓝色

sum(c,b-1):绿色+蓝色

sum(a-1,d):绿色+黄色

sum(a-1,b-1):绿色

则红色区域元素之和=sum(c,d)-sum(c,b-1)-sum(a-1,d)+sum(a-1,b-1)

画个图就可以得到 。

#include <bits/stdc++.h>
using namespace std;
const int maxn = 1105;
int bit[maxn][maxn];

void updata(int x,int y,int val)
{
for(int i=x;i<maxn;i+=(i &(-i)))
{
for(int j=y;j<maxn ; j+=(j & (-j)))
{
bit[i][j] += val;
}
}
}
int query(int x,int y)
{
int ans=0;
for(int i=x; i>0 ;i-=(i&(-i)))
{
for(int j=y;j>0;j-=(j&(-j)))
{
ans+=bit[i][j];
}
}
return ans;
}
int main()
{
int x1,x2,y1,y2,T,kiss=0,n,val;
scanf("%d",&T);
char s[30];
while(T--)
{
for(int i=1;i<maxn;++i)
{
for(int j=1;j<maxn;++j)
{
bit[i][j] = 0 ;
}
}
scanf("%d",&n);
printf("Case %d:\n",++kiss);

for(int i=1;i<maxn;++i)
{
for(int j=1;j<maxn ;++j)
{
updata(i,j,1);
}
}

while(n--)
{
scanf("%s",s);
if(s[0]=='S')
{
scanf("%d%d%d%d",&x1,&y1,&x2,&y2);
x1++; x2++; y1++; y2++;

if(x1<x2) swap(x1,x2);
if(y1<y2) swap(y1,y2);
int sum = query(x1,y1) - query(x2-1,y1) - query(x1,y2-1) + query(x2-1,y2-1);
printf("%d\n",sum);
}
else if(s[0]=='A')
{
scanf("%d%d%d",&x1,&y1,&val);
x1++; y1++;
updata(x1,y1,val);
}
else if(s[0]=='D')
{
scanf("%d%d%d",&x1,&y1,&val);
x1++; y1++;
int sum = query(x1,y1) - query(x1-1,y1) - query(x1,y1-1) + query(x1-1,y1-1);
int temp = min(sum,val);
updata(x1,y1,-temp);
}
else
{
scanf("%d%d%d%d%d",&x1,&y1,&x2,&y2,&val);
x1++; y1++; x2++; y2++;
int sum = query(x1,y1) - query(x1-1,y1) - query(x1,y1-1) + query(x1-1,y1-1);
int temp = min(sum,val);
updata(x1,y1,-temp);
updata(x2,y2,temp);
}
}

}
return 0;
}