【poj2155】Matrix 樹狀數組

題目大意

給定一個N*N的矩陣a,a矩陣的初值設爲0。T次操作,涉及區間翻轉0和1以及單點查詢。
2 <= N <= 1000, 1 <= T <= 50000

題解

我的做法好像跟其他人不太一樣qwq
用二維樹狀數組維護差分,每次(x1,y1)+1,(x2+1,y1)-1,(x1,y2+1),因爲右下部分多減了了一次,所以(x2+1,y2+1)+1,然後詢問的時候只要加起來然後看是偶數還是奇數即可。

代碼

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
#include<cctype>
using namespace std;
int c[1010][1010];
int T,n,m;
int lowbit(int x){return x&(-x);}
void update(int x,int y,int z)
{
	for (int i=x;i<=n;i+=lowbit(i))
		for (int j=y;j<=n;j+=lowbit(j)) c[i][j]+=z;
}
int query(int x,int y)
{
	int ans=0;
	for (int i=x;i>0;i-=lowbit(i))
		for (int j=y;j>0;j-=lowbit(j)) ans+=c[i][j];
	return ans&1?1:0;
}
int main()
{
	for (scanf("%d",&T);T;T--)
	{
		memset(c,0,sizeof(c));
		scanf("%d%d",&n,&m);;
		for (int i=1;i<=m;i++)
		{
			int x1,x2,y1,y2;
			char ch;scanf(" %c",&ch);
			if (ch=='C')
			{
				scanf("%d%d%d%d",&x1,&y1,&x2,&y2);
				update(x1,y1,1);update(x2+1,y2+1,1);
				update(x2+1,y1,-1);update(x1,y2+1,-1);
			}
			else
				scanf("%d%d",&x1,&y1),printf("%d\n",query(x1,y1));
		}
		printf("\n");
	}
	return 0;
}

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章