通过这两天对树状数组的学习,感觉树状数组非常强大,在统计记数中效率很高。不仅可以统计一维的还可以统计二维的,甚至三维的。。。。树状数组能解决的题一般有。统计数组中有多少元素。。统计有多少比给定数大的元素,,,统计在一个矩形区域内有多少元素。。。
这一题的题意是:可以把星星看做灯泡,给个一些灯泡的位置坐标,在一个矩形区域内让你统计;有多少亮着的灯泡,,。。。
思路:初始化一开始的灯泡都不亮。。当给定一个坐标时,如果该坐标的灯泡是亮的。如果遇见D就关闭该灯泡,并更新相应位置的值。如果一开始是灭的,如果遇见B就打开该灯泡,并更新相应的值,仅在这两种情况时才更新。。。
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<string.h>
#define N 1005
using namespace std;
int s[N][N];
bool flag[N][N];
int lowbit(int x)
{return x&(-x);}
void update(int x,int y,int v)
{ for(int i=x;i<N;i+=lowbit(i))
for(int j=y;j<N;j+=lowbit(j))
s[i][j]+=v;
}
int Quary(int x,int y)
{ int sum=0;
for(int i=x;i>0;i-=lowbit(i))
for(int j=y;j>0;j-=lowbit(j))
sum+=s[i][j];
return sum;
}
int main()
{ int n;
cin>>n;
char ch[10];
memset(s,0,sizeof(s));
memset(flag,false,sizeof(flag));
int a,b,c,d;
for(int i=0;i!=n;++i)
{ scanf("%s",ch);
if(ch[0]=='B')
{ scanf("%d%d",&a,&b);
if(!flag[a+1][b+1])
{ flag[a+1][b+1]=true;
update(a+1,b+1,1);
}
}
else if(ch[0]=='D')
{ scanf("%d%d",&a,&b);
if(flag[a+1][b+1])
{ flag[a+1][b+1]=false;
update(a+1,b+1,-1);
}
}
else { scanf("%d%d%d%d",&a,&b,&c,&d);
if(a>b) swap(a,b);
if(c>d) swap(c,d);
int res=Quary(b+1,d+1)-Quary(a,d+1)-Quary(b+1,c)+Quary(a,c);
printf("%d\n",res);
}
}return 0;
}