poj 2155 Matrix(二维树状数组)

题目大意：

有一个N*N的数组a，初始值为1

分为两种操作：

操作一：将所有满足x1<=i<=x2 且 y1<=j<=y2的a[i][j]反置，即0->1，1->0

操作二，询问a[x][y]是什么。

 

可以将反置变成所有区域每一个数都加一，最后将（a[x][y]%2）输出

这一题是用二维树状数组更新区间、单点查询的题。

对于操作一，进行如下维护：

h[x1][y1] ++; h[x1][y2+1] --; h[x2+1][y1]--; h[x2+1][y2+1]++;

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;
}

if (c == 'C')
{
	scanf("%d%d%d%d",&x1,&y1,&x2,&y2);
	updata(x1, y1, 1);
	updata(x1, y2+1, -1);
	updata(x2+1, y1, -1);
	updata(x2+1, y2+1, 1);
}

 

对于操作二，查询x,y的值

h[1][1] + h[1][2] + ..... + h[1][y] +

......

+h[x][1] + h[x][2] + ...... + h[x][y]

即是答案。

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;
}

else if (c == 'Q')
{
	scanf("%d%d",&x,&y);
	ans = query(x, y);
	ans %= 2;
	printf("%d\n", ans);
}

 

这些都可以用二维树状数组维护和查询

 

我的全部代码：

#include <iostream>
#include <stdio.h>
#include <stdlib.h>
using namespace std;

const int MAXN = 1010;

int h[MAXN][MAXN];
int n, m;

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 T, i, j, ans;
	int x, y, x1, y1, x2, y2;
	char c;
	scanf("%d",&T);
	while (T--)
	{
		scanf("%d%d",&n,&m);
		for (i=1; i<=n; i++)
			for (j=1; j<=n; j++)
				h[i][j] = 0;
		for (i=1; i<=m; i++)
		{
			scanf("%c",&c);
			scanf("%c",&c);
			if (c == 'C')
			{
				scanf("%d%d%d%d",&x1,&y1,&x2,&y2);
				updata(x1, y1, 1);
				updata(x1, y2+1, -1);
				updata(x2+1, y1, -1);
				updata(x2+1, y2+1, 1);
			}
			else if (c == 'Q')
			{
				scanf("%d%d",&x,&y);
				ans = query(x, y);
				ans %= 2;
				printf("%d\n", ans);
			}
		}
		if (T != 0)
			printf("\n");
	}
	return 0;
}