可以理解成一个强化版的求最大子矩阵
四种操作
S x1 y1 x2 y2 询问以(x1 , y1) - (x2 , y2)为对角线的矩形的面积,但是这个对角线不一定是正对角线。
A x1 y1 n 把点(x1 , y1)加上n。
D x1 y1 n点(x1 , y1)减去n如果不足n就全部删除即可。
M x1 y1 x2 y2 n 把点(x1 , y1)点值中扣除n加到(x2 , y2),如果不足n则把(x1 , y1)值全部加到(x2 , y2)
如果用暴力方法会出现超时的可能,本题就是一个加入各种骚操作的矩阵和,如果直接求矩阵和,我们完全可以用前缀和解决,但是问题在于,这道题加入了修改操作。。
这么一想不就和当初刚刚接触线段树和树状数组时很像吗,当初只是在求一维空间的某段和以及修改,现在改成二维空间,那么我们完全可以用求矩阵权值和的种种思路,用二维的树状数组对修改相加的过程进行优化。
先把全部矩阵赋值为 1 ,而且在开一个前缀和数组基础上保留一下原数组用来特判 D M 操作是否加上了一个越界的数字,还有 S 操作要注意给的矩阵的坐标是不是一个正对角线。。
很优秀的一道题。。。
以下是 AC 代码
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn = 1005;
const int low(int x){return x&-x;}
int tree[maxn][maxn];
int num[maxn][maxn];
void add(int x,int y,int w)
{
for(int i=x;i<maxn;i+=low(i))
{
for(int j=y;j<maxn;j+=low(j))
{
tree[i][j] += w;
}
}
}
int sum(int x,int y)
{
int res = 0;
for(int i=x;i>0;i-=low(i))
{
for(int j=y;j>0;j-=low(j))
{
res += tree[i][j] ;
}
}
return res;
}
void init()
{
memset(tree,0,sizeof tree);
for(int i=1;i<maxn;i++)
for(int j=1;j<maxn;j++)
{
tree[i][j] = low(i)*low(j);
num[i][j] = 1;
}
}
int main()
{
int t,n,ct = 1;
scanf("%d",&t);
while(t--)
{
scanf("%d",&n);
init();
printf("Case %d:\n", ct++);
int x,y,val;
int x1,y1, x2,y2;
int x3,y3, x4,y4;
char ch[4];
while(n--)
{
cin >> ch;
if(ch[0] == 'S')
{
scanf("%d%d%d%d",&x1,&y1,&x2,&y2);
x1++,y1++,x2++,y2++;
x3=min(x1,x2), x4=max(x1,x2);
y3=min(y1,y2), y4=max(y1,y2);
int ans = sum(x4,y4);
ans -= sum(x3-1,y4);
ans -= sum(x4,y3-1);
ans += sum(x3-1,y3-1);
printf("%d\n",ans);
}
else if(ch[0] == 'A')
{
scanf("%d%d%d",&x,&y,&val);
x++, y++;
add(x,y,val);
num[x][y] += val;
}
else if(ch[0] == 'D')
{
scanf("%d%d%d",&x,&y,&val);
x++, y++;
val = min(val, num[x][y]);
add(x,y,-val);
num[x][y] -= val;
}
else
{
scanf("%d%d%d%d%d",&x1,&y1,&x2,&y2,&val);
x1++,y1++,x2++,y2++;
val = min(val,num[x1][y1]);
num[x1][y1] -= val;
num[x2][y2] += val;
add(x1,y1,-val);
add(x2,y2,val);
}
}
}
return 0;
}