通过这两天对树状数组的学习,感觉树状数组非常强大,在统计记数中效率很高。不仅可以统计一维的还可以统计二维的,甚至三维的。。。。树状数组能解决的题一般有。统计数组中有多少元素。。统计有多少比给定数大的元素,,,统计在一个矩形区域内有多少元素。。。
这一题的题意是:可以把星星看做灯泡,给个一些灯泡的位置坐标,在一个矩形区域内让你统计;有多少亮着的灯泡,,。。。
思路:初始化一开始的灯泡都不亮。。当给定一个坐标时,如果该坐标的灯泡是亮的。如果遇见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; }