数算实习 Mobile phones 树状数组

Mobile phones
描述
假设某地区的第四代移动电话基站可以接收手机信号。该区域为正方形，形成一个S * S大小的矩阵，行和列的编号从0到S-1。每个方块包含一个基站。手机内的有源手机数量可能会发生变化，因为手机从可以在方块间任意移动或开机关机。

编写一个程序，接收这些报告并回答任何矩形区域中当前活动手机的总数。

输入：
输入包含两部分：instruction以及其代表的含义
当instruction=0时，初始化区域矩阵使之全为0，并输入n，表示矩阵大小
当instruction=1时，输入x,y,A，表示（x,y）处加入A个手机
当instruction=2时，输入x1,y1,x2,y2，并输出（x1,y1）至（x2,y2）之间整个区域的手机数

样例输入
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

#include <iostream>
using namespace std;

int c[1025][1025];
int n;
int lowbit(int x)
{
	return x & -x;
}

void update(int x, int y, int delta)               //更新树状数组
{
	for (int i = x; i <= n; )
	{
		for (int j = y; j <= n;)
		{
			c[i][j] += delta;
			j = j + lowbit(j);
		}
		i = i + lowbit(i);
	}
}

int calculate(int x, int y)                         //求和
{
	int sum = 0;
	for (int i = x; i > 0;)
	{
		for (int j = y; j > 0;)
		{
			sum += c[i][j];
			j = j - lowbit(j);
		}
		i = i - lowbit(i);
	}
	return sum;
}

int main()
{
	int ins;
	while (1)
	{
		cin >> ins;
		if (ins == 3)
			break;
		if (ins == 0)
		{
			cin >> n;
			for (int i = 0; i <= n; i++)
				for (int j = 0; j <= n; j++)
					c[i][j] = 0;
			continue;
		}
		if (ins == 1)
		{
			int x, y, A;
			cin >> x >> y >> A;
			update(x + 1, y + 1, A);
		}
		else if (ins == 2)
		{
			int x1, y1, x2, y2;
			cin >> x1 >> y1 >> x2 >> y2;
			int result;
			result = calculate(x2 + 1, y2 + 1) + calculate(x1, y1) - calculate(x1, y2 + 1) - calculate(x2 + 1, y1);
			cout << result << endl;
		}
	}
	return 0;
}