[BZOJ3132]上帝造题的七分钟

来源:互联网 发布:网络映射 编辑:程序博客网 时间:2024/04/30 01:54

原题地址

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

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

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