poj 1656（二维树状数组解法）

题目信息：count black

利用二维树状数组：

//利用二维树状数组 
#include<iostream>
#include<string>
#include<cstring>
using namespace std;

int c[101][101];
int b[101][101];
int Row,Col;

//下面三个函数是基本套路 
inline int Lowbit(const int &x)
{
    return x&(-x);
}

int Sum(int i,int j)
{//计算(1,1)--(i,j)之间的和 
    int tempj,sum=0;
	while(i > 0)
	{
		tempj=j;
		while(tempj > 0){
		    sum+=c[i][tempj];
			tempj-=Lowbit(tempj);
		}
		i-=Lowbit(i);
	}
	return sum;
}

void Update(int i,int j,int num)
{//更新(i,j) 为c[i][j]+num 
    int tempj;
	while(i<=Row)
	{
	    tempj=j;
		while(tempj<=Col){
		    c[i][tempj]+=num;
			tempj+=Lowbit(tempj);
		}
		i+=Lowbit(i);
	}
}

void whiteUpdate(int x,int y,int L)
{
    for(int i=x;i<x+L;++i)
		for(int j=y;j<y+L;++j)
		{
		    if(b[i][j]==1) {b[i][j]=0;Update(i,j,-1);}
			else continue;
		}
}

void blackUpdate(int x,int y,int L)
{
    for(int i=x;i<x+L;++i)
		for(int j=y;j<y+L;++j)
		{
		    if(b[i][j]==1) continue;
			else {b[i][j]=1;Update(i,j,1);};
		}
}

int blackSum(int x,int y,int L)
{//计算(x,y)--(x+L-1,Y+L-1)之间的数 
   return Sum(x+L-1,y+L-1)+Sum(x-1,y-1)-Sum(x-1,y+L-1)-Sum(x+L-1,y-1); 
}

int main()
{
    int x,y,L,N;
	string str;
	Row=Col=100;
	memset(c,0,sizeof(c));
	memset(b,0,sizeof(b));
	cin>>N;
	while(N--)
	{
	    cin>>str>>x>>y>>L;
		switch(str[0])
		{
		    case 'W':
				whiteUpdate(x,y,L);
				break;
			case 'B':
				blackUpdate(x,y,L);
				break;
			case 'T':
				cout<<blackSum(x,y,L)<<endl;
				break;
			default: break;
		}
	}
	return 0;
}