[BZOJ3132]上帝造题的七分钟

原题地址

维护一个矩阵,支持范围加和范围求和.

据说二维线段树做不了,用二维树状数组类似一维的差分序列技巧处理即可.

AC code:

#include <cstdio>
const int N=2049;
int  n,m,a,b,c,d,v;
int  D[N][N],ID[N][N],JD[N][N],IJD[N][N];
char t;

void add(int T[][N],int x,int y,int v){
    for(int i=x;i<=n;i+=i&-i)
        for(int j=y;j<=m;j+=j&-j) T[i][j]+=v;
}

int get(int T[][N],int x,int y){
    int sum=0;
    for(int i=x;i;i-=i&-i)
        for(int j=y;j;j-=j&-j) sum+=T[i][j];
    return sum;
}

void Add(int x,int y,int v){
    add(D,x,y,v);
    add(ID,x,y,x*v);
    add(JD,x,y,y*v);
    add(IJD,x,y,x*y*v);
}

int Get(int x,int y){
    return (x*y+x+y+1)*get(D,x,y)-(y+1)*get(ID,x,y)-(x+1)*get(JD,x,y)+get(IJD,x,y);
}

int main(){
    scanf("%c %d%d",&t,&n,&m);
    while(scanf("\n%c ",&t)!=EOF){
        if(t=='L'){
            scanf("%d%d%d%d%d",&a,&b,&c,&d,&v);
            Add(a,b,v);Add(a,d+1,-v);Add(c+1,b,-v);Add(c+1,d+1,v);
        }
        else{
            scanf("%d%d%d%d",&a,&b,&c,&d);
            printf("%d\n",Get(c,d)-Get(c,b-1)-Get(a-1,d)+Get(a-1,b-1));
        }
    }

    return 0;
}