poj 1195 Mobile phones (二维树状数组)

题目大意：

有一个S*S的数组。

有两种操作：

一、h[X,Y]加上A

二、查询所有x1<=i<=x2, y1<=j<=y2的h[i, j]的和。

 

这一题是一道二维树状数组的模板体。

如果要查询（x1,y1,x2,y2）的值：只要查询query(1, 1, x2, y2) + query(1, 1, x1-1, y1-1) - query(1, 1, x2, y1-1) - query(1, 1, x1-1, y2)

这个查询和维护可以使用二维树状数组来维护

 

我的代码：

#include <iostream>
#include <stdio.h>
#include <stdlib.h>
#include <string>
#include <cstring>
using namespace std;
const int MAXN = 1050;
int h[1050][1050];
int n;

int lowbit(int x)
{
	return x & (-x);
}
void updata(int x, int y, int val)
{
	int i, j;
	for (i=x; i<=n; i+=lowbit(i))
		for (j=y; j<=n; j+=lowbit(j))
			h[i][j] += val;
}
int query(int x, int y)
{
	int i, j, ans = 0;
	for (i=x; i>0; i-=lowbit(i))
		for (j=y; j>0; j-=lowbit(j))
			ans += h[i][j];
	return ans;
}
int main()
{
	int tt, x, y, a, x1, y1, x2, y2, ans;
	scanf("%d%d",&tt,&n);
	memset(h, 0, sizeof(h));
	while (scanf("%d",&tt))
	{
		if (tt == 1)
		{
			scanf("%d%d%d",&x,&y,&a);
			x++; y++;
			updata(x, y, a);
		}
		else if (tt == 2)
		{
			scanf("%d%d%d%d",&x1,&y1,&x2,&y2);
			x2++; y2++; x1++; y1++;
			ans = query(x2,y2) + query(x1-1, y1-1) - query(x2, y1-1) - query(x1-1, y2);
			printf("%d\n", ans);
		}
		else break;
	}
	return 0;
}