【BZOJ 3132】上帝造题的七分钟【二维树状数组】

Description

“第一分钟，X说，要有矩阵，于是便有了一个里面写满了0的n×m矩阵。
第二分钟，L说，要能修改，于是便有了将左上角为(a,b)，右下角为(c,d)的一个矩形区域内的全部数字加上一个值的操作。
第三分钟，k说，要能查询，于是便有了求给定矩形区域内的全部数字和的操作。
第四分钟，彩虹喵说，要基于二叉树的数据结构，于是便有了数据范围。
第五分钟，和雪说，要有耐心，于是便有了时间限制。
第六分钟，吃钢琴男说，要省点事，于是便有了保证运算过程中及最终结果均不超过32位有符号整数类型的表示范围的限制。
第七分钟，这道题终于造完了，然而，造题的神牛们再也不想写这道题的程序了。”
——《上帝造裸题的七分钟》
所以这个神圣的任务就交给你了。

Input

输入数据的第一行为X n m，代表矩阵大小为n×m。
从输入数据的第二行开始到文件尾的每一行会出现以下两种操作：
　　L a b c d delta —— 代表将(a,b),(c,d)为顶点的矩形区域内的所有数字加上delta。
　　k a b c d　　 —— 代表求(a,b),(c,d)为顶点的矩形区域内所有数字的和。
请注意，k为小写。

Output

针对每个k操作，在单独的一行输出答案。

Sample Input

X 4 4
L 1 1 3 3 2
L 2 2 4 4 1
k 2 2 3 3

Sample Output

12

HINT

对于100%的数据，1 ≤ n ≤ 2048, 1 ≤ m ≤ 2048, 1 ≤ abs(delta) ≤ 500,操作不超过200000个,保证运算过程中及最终结果均不超过32位带符号整数类型的表示范围。

题解

用类似一维树状数组的方法
详见代码

代码

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;

#define N 2050
int n,m;
struct BIT{
    int f[N][N];
    int lowbit(int x) {return x & -x;}
    inline void add(int x,int y,int v)
    {
        for(int i = x;i <= n;i += lowbit(i))
            for(int j = y;j <= m;j += lowbit(j))
                f[i][j] += v;
    }
    inline int query(int x,int y)
    {
        int res = 0;
        for(int i = x;i >= 1;i -= lowbit(i))
            for(int j = y;j >= 1;j -= lowbit(j))
                res += f[i][j];
        return res;
    }
}a,b,c,d;

void work(int x1,int y1,int x2,int y2,int v)
{
    a.add(x1,y1,v);a.add(x2+1,y2+1,v);
    a.add(x2+1,y1,-v);a.add(x1,y2+1,-v);

    b.add(x1,y1,x1*v);b.add(x2+1,y2+1,(x2+1)*v);
    b.add(x2+1,y1,-(x2+1)*v);b.add(x1,y2+1,-x1*v);

    c.add(x1,y1,y1*v);c.add(x2+1,y2+1,(y2+1)*v);
    c.add(x2+1,y1,-y1*v);c.add(x1,y2+1,-(y2+1)*v);

    d.add(x1,y1,x1*y1*v);d.add(x2+1,y2+1,(x2+1)*(y2+1)*v);
    d.add(x2+1,y1,-(x2+1)*y1*v);d.add(x1,y2+1,-x1*(y2+1)*v);
}

int sum(int x,int y)
{
    return (x+1)*(y+1)*a.query(x,y)-(y+1)*b.query(x,y)-(x+1)*c.query(x,y)+d.query(x,y);
}

int main()
{
    char ch[3];
    int x1,x2,y1,y2,v;
    scanf("%s",ch);
    while(ch[0] != 'X') scanf("%s",ch);
    scanf("%d%d",&n,&m);
    while(scanf("%s",ch) != EOF)
    {
        scanf("%d%d%d%d",&x1,&y1,&x2,&y2);
        if(ch[0] == 'L')
        {
            scanf("%d",&v);
            work(x1,y1,x2,y2,v);
        }
        else
            printf("%d\n",sum(x2,y2) - sum(x1-1,y2) - sum(x2,y1-1) + sum(x1-1,y1-1));
    }
    return 0;
}