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;}