HDU 1892 See you~ (二维树状数组)

题目链接: 戳我

题目大意:

给你一个书架, 1000*1000的矩阵,即格子.每个格子初始化都是1, 有下面几种操作:

S x1 y1 x2 y2 意思是你应该告诉我 以（X1，Y1）,（X2，Y2）作为对角线，包括两点的矩形的总共有多少书。坑点: (x1, y1) 这个点不一定比 (x2, y2)这个点的小
A x1 y1 n1 意思是在(x1, y1) 这个位置放 n1 本书

D x1 y1 n1 意思是在 位置（X1，Y1）拿走 n1 本书，如果在那个位置上的书低于n1，则搬开所有的书。 反正不能出现负数
M x1 y1 x2 y2 n1 把(x1, y1)这个位置的 n1 本书 移动到 (x2, y2) 这个位置. 如果在那个位置上的书低于n1，则移走所有的书

注意: 0<=x1,y1,x2,y2<=1000,1<=n1<=100.

简单的二维树状数组.

//Author LJH
//www.cnblogs.com/tenlee
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <cctype>
#include <cmath>
#include <algorithm>
#include <vector>
#include <queue>
#include <stack>
#include <map>
#define clc(a, b) memset(a, b, sizeof(a))
using namespace std;

const int inf = 0x3f;
const int INF = 0x3f3f3f3f;
const int maxn = 1e3+5;

int a[maxn][maxn];

int lowbit(int x)
{
return x & -x;
}

void Add(int x, int y, int val)
{
int yy = y;
while(x < maxn)
{
yy = y;
while(yy < maxn)
{
a[x][yy] += val;
yy += lowbit(yy);
}
x += lowbit(x);
}
}

int Sum(int x, int y)
{
int yy = y, ans = 0;
while(x)
{
yy = y;
while(yy)
{
ans += a[x][yy];
yy -= lowbit(yy);
}
x -= lowbit(x);
}
return ans;
}

void Init()
{
clc(a, 0);
for(int i = 1; i < maxn; i++)
{
for(int j = 1; j < maxn; j++)
{
Add(i, j, 1);
}
}
}

int main()
{
int T, q, x1, y1, x2, y2, n1, sum;
char op;
scanf("%d", &T);
for(int t = 1; t <= T; t++)
{
scanf("%d", &q);
Init();
printf("Case %d:\n", t);
while(q--)
{
scanf(" %c", &op);
if(op == 'S')
{
scanf("%d %d %d %d", &x1, &y1, &x2, &y2);
x1++, y1++, x2++, y2++;
int a1, b1, a2, b2;
a1 = min(x1,x2); a2 = max(x1,x2);
b1 = min(y1,y2); b2 = max(y1,y2);
printf("%d\n",Sum(a2,b2) - Sum(a1-1,b2) - Sum(a2,b1-1) + Sum(a1-1,b1-1));
}
else if ( op == 'A' )
{
scanf("%d %d %d", &x1, &y1, &n1);
Add(x1+1, y1+1, n1);
}
else if( op == 'D' )
{
int temp;
scanf("%d %d %d", &x1, &y1, &n1);
x1++, y1++;
temp = Sum(x1,y1) - Sum(x1-1,y1) - Sum(x1,y1-1) + Sum(x1-1,y1-1);
if(temp >= n1)
Add(x1, y1, -n1);
else
Add(x1, y1, -temp);
}
else if(op == 'M')
{
scanf("%d %d %d %d %d", &x1, &y1, &x2, &y2, &n1);
x1++, y1++, x2++, y2++;
int temp = Sum(x1,y1) - Sum(x1-1,y1) - Sum(x1,y1-1) + Sum(x1-1,y1-1);
if(temp >= n1)
{
Add(x1, y1, -n1);
Add(x2, y2, n1);
}
else
{
Add(x1, y1, -temp);
Add(x2, y2, temp);
}
}
}
}
return 0;
}