描述:
假设南京鼓楼区的移动电话基站为如下操作:把该区域分成一个正方形,形成一个 S * S 的 矩阵,行和列的编号为从 0 到 S-1。正方形的每个块为一个基站。 基站内的活跃手机数量 可能会发生变化,因为手机会从一个块移动到另一个块,或者开机和关机。 在这时,每个 基站会向主基站报告活跃手机数量的变化以及矩阵变化位置的行和列。
编写一个程序,接收这些报告,并回答有关任何矩阵中当前活跃手机总数的查询。
输入:
每一次输入会在单独的一行上,包含一个整数指令和多个整数参数。具体输入如下:
输入值可以考虑为始终在范围内,对于 A 是负数而言,可以保证它不会把矩阵值减小到负数。索引查询从 0 开始,例如,对于大小为 4 * 4 的表,我们有 0<=X<=3 且 0<=Y<=3。
表的大小:1 * 1 <= S * S <= 1024 * 1024
每个基站中的值 V(任何时间):0 <= V <= 32767
更新的活跃手机数量:-32768 <= A <= 32767 输入的指令数:3 <= U <= 60002
整个表中的最大电话数:M = 2 ^ 30
输出:
您的程序不应该对除 2 之外的指令的行返回任何内容。如果指令是 2,那么您的程序应该通 过将答案写为包含单个整数的标准输出。
样例输入:
0 4
1 1 2 3
2 0 0 2 2
1 1 1 2
1 1 2 -1
2 1 1 2 3
3
样例输出:
3
4
思路:一个二维树状数组的板子题。树状数组的话,这个博客讲的很清楚,放个传送门~这道题需要注意的是下标是从0开始的,所以在具体操作前要先给下标+1。
代码:
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <cmath>
#include <iostream>
#include <algorithm>
#include <string>
#include <vector>
#include <queue>
#include <map>
#include <set>
using namespace std;
int op, s, x, y, a, l, b, r, t;
int C[1030][1030] = { 0 };
int lowbit(int x)
{
return x & (-x);
}
void update(int x, int y, int val)
{
for (int i = x; i <= s; i += lowbit(i))
{
for (int j = y; j <= s; j += lowbit(j))
C[i][j] += val;
}
}
int sum(int x, int y)
{
int res = 0;
for (int i = x; i > 0; i -= lowbit(i))
{
for (int j = y; j > 0; j -= lowbit(j))
res += C[i][j];
}
return res;
}
int main()
{
while (cin >> op)
{
if (op == 3)
break;
if (op == 0)
cin >> s;
else if (op == 1)
{
cin >> x >> y >> a;
x++, y++;
update(x, y, a);
}
else if (op == 2)
{
cin >> l >> b >> r >> t;//(l,b)~(r,t)
l++, b++, r++, t++;
cout<< sum(r, t) - sum(r, b - 1) - sum(l - 1, t) + sum(l - 1, b - 1) << endl;
}
}
return 0;
}