【bzoj3132】上帝造题的七分钟 二维树状数组区间修改区间查询

时间:2021-10-13 15:37:52

题目描述

“第一分钟,X说,要有矩阵,于是便有了一个里面写满了0的n×m矩阵。

第二分钟,L说,要能修改,于是便有了将左上角为(a,b),右下角为(c,d)的一个矩形区域内的全部数字加上一个值的操作。

第三分钟,k说,要能查询,于是便有了求给定矩形区域内的全部数字和的操作。

第四分钟,彩虹喵说,要基于二叉树的数据结构,于是便有了数据范围。

第五分钟,和雪说,要有耐心,于是便有了时间限制。

第六分钟,吃钢琴男说,要省点事,于是便有了保证运算过程中及最终结果均不超过32位有符号整数类型的表示范围的限制。

第七分钟,这道题终于造完了,然而,造题的神牛们再也不想写这道题的程序了。”

——《上帝造裸题的七分钟》

所以这个神圣的任务就交给你了。

输入

输入数据的第一行为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为小写。

输出

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

样例输入

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

样例输出

12


题解

二维树状数组区间修改区间查询

头一次知道树状数组还能实现区间修改区间查询~

我们先考虑一维的情况:树状数组的本质是$O(\log n)$维护前缀和,前缀相减可以得到区间和。

而又有一种区间修改单点查询的方法:差分。

设$d[i]=a[i]-a[i-1]$,那么$a[i]=(a[i]-a[i-1])+(a[i-1]-a[i-2])+...+(a[1]-a[0])=\sum\limits_{i=1}^nd[i]$,于是可以按照相同的方法维护差分数组的前缀和,修改的时候直接将l加上,r+1减去即可。

那么对于区间查询呢?我们还是引入差分的思想,则$\sum\limits_{i=1}^na[i]=\sum\limits_{i=1}^n\sum\limits_{j=1}^id[j]=\sum\limits_{j=1}^n(n+1-j)d[j]=(n+1)\sum\limits_{j=1}^nd[j]-\sum\limits_{j=1}^nj·d[j]$。所以维护两个树状数组,一个维护d[j]的前缀和,一个维护j*d[j]的前缀和就可以了。

那么对于本题变成二维的呢?其实也是一样的。

还是使用差分的方法,$a[n][m]=\sum\limits_{i=1}^n\sum\limits_{j=1}^md[i][j]$,$\sum\limits_{i=1}^n\sum\limits_{j=1}^ma[i][j]=\sum\limits_{i=1}^n\sum\limits_{j=1}^m\sum\limits_{k=1}^i\sum\limits_{l=1}^jd[k][l]=\sum\limits_{k=1}^n\sum\limits_{l=1}^m(n+1-k)(m+1-l)d[k][l]$。

展开什么的,自己动手丰衣足食

然后开4个树状数组即可。修改和查询时使用容斥原理拆成4段来完成。

#include <cstdio>
#include <algorithm>
#define N 2055
using namespace std;
int n , m;
char str[5];
struct data
{
int f[N][N];
void update(int x , int y , int a)
{
int i , j;
for(i = x ; i <= n ; i += i & -i)
for(j = y ; j <= m ; j += j & -j)
f[i][j] += a;
}
int query(int x , int y)
{
int i , j , ans = 0;
for(i = x ; i ; i -= i & -i)
for(j = y ; j ; j -= j & -j)
ans += f[i][j];
return ans;
}
}A , B , C , D;
void modify(int x , int y , int z)
{
A.update(x , y , z) , B.update(x , y , x * z) , C.update(x , y , y * z) , D.update(x , y , x * y * z);
}
int solve(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()
{
int x1 , y1 , x2 , y2 , z;
scanf("%*s%d%d" , &n , &m);
while(~scanf("%s%d%d%d%d" , str , &x1 , &y1 , &x2 , &y2))
{
if(str[0] == 'L')
{
scanf("%d" , &z) , x2 ++ , y2 ++ ;
modify(x1 , y1 , z) , modify(x2 , y1 , -z) , modify(x1 , y2 , -z) , modify(x2 , y2 , z);
}
else x1 -- , y1 -- , printf("%d\n" , solve(x1 , y1) - solve(x2 , y1) - solve(x1 , y2) + solve(x2 , y2));
}
return 0;
}