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