二维树状数组

Mobile phones http://poj.org/problem?id=1195

#include<cstdio>
#include<cstring>
#define mt(a,b) memset(a,b,sizeof(a))
const int M=1030;
class Two_Tree_Array {  // 二维树状数组
    typedef int typev;
    typev c[M][M];
public:
    void init() {
        mt(c,0);
    }
    int lowb(int t) {
        return t&(-t);
    }
    void add(int x,int y,typev v) {
        for(int i=x; i<M; i+=lowb(i))
            for(int j=y; j<M; c[i][j]+=v,j+=lowb(j));
    }
    typev sum(int x,int y) {
        typev res=0;
        for(int i=x; i>0; i-=lowb(i))
            for(int j=y; j>0; res+=c[i][j],j-=lowb(j));
        return res;
    }
    typev query(int x1,int y1,int x2,int y2) {
        return sum(x2,y2)-sum(x2,y1-1)-sum(x1-1,y2)+sum(x1-1,y1-1);
    }
} gx;
int main(){
    int op,l,b,r,t,n;
    while(~scanf("%d",&op),op!=3){
        if(!op){
            scanf("%d",&n);
            gx.init();
        }
        else if(op&1){
            scanf("%d%d%d",&l,&b,&op);
            gx.add(l+1,b+1,op);
        }
        else{
            scanf("%d%d%d%d",&l,&b,&r,&t);
            printf("%d\n",gx.query(l+1,b+1,r+1,t+1));
        }
    }
    return 0;
}

 

 

 

 

end