POJ2155 Matrix 二维树状数组 修改区域，查询节点

Problem Address：http://poj.org/problem?id=2155

 

【前言】

 

参考资料：http://blog.csdn.net/Human_CK/archive/2011/06/13/6542370.aspx

 

昨天学习了树状数组，今天就是实战的时候啦！

 

马上找了一道典型的题目做。

 

写了但忘记调用初始化函数而WA了一次。

 

不过这么快就过来还是蛮兴奋的。

 

【思路】

 

这道题是简单的“改段求点”，直接套用模板。

 

要注意的是，这道题是翻转而不是增加某个值。

 

事实上可以把它看成增加1，求结果的时候模2就行了。

 

而这里其实可以用一个小技巧。

 

比如某个点为x。x不是0就是1。那么翻转使用公式x = 1 - x。这样就可以轻易翻转了。

 

二维的树状数组只要对一维的情况加多一个循环就OK啦~

 

【代码】

 

#include <iostream> using namespace std; const int maxn = 1000; int a[maxn+10][maxn+10]; int b[maxn+10][maxn+10]; inline int lowbit(int x) { return x & (-x); } void init(int n) { int i,j; for (i=1; i<=n; i++) { for (j=1; j<=n; j++) { a[i][j] = 0; b[i][j] = 0; } } } inline void update_B(int n, int x, int y) { int i,j; for (i=x; i>0; i-=lowbit(i)) { for (j=y; j>0; j-=lowbit(j)) { b[i][j] = 1 - b[i][j]; } } } inline void update(int n, int x1, int y1, int x2, int y2) { update_B(n, x2, y2); update_B(n, x2, y1-1); update_B(n, x1-1, y2); update_B(n, x1-1, y1-1); } inline int sum(int n, int x, int y) { int i,j,s=0; for (i=x; i<=n; i+=lowbit(i)) { for (j=y; j<=n; j+=lowbit(j)) { s += b[i][j]; } } return s%2; } int main() { int index = 0; int x; int n,t; char c[3]; int x1,y1,x2,y2; scanf("%d", &x); while(x--) { scanf("%d %d", &n, &t); if (index==1) printf("/n"); index = 1; init(n); while(t--) { scanf("%s", c); if (c[0]=='C') { scanf("%d %d %d %d", &x1, &y1, &x2, &y2); update(n, x1, y1, x2, y2); } else { scanf("%d %d", &x1, &y1); printf("%d/n", sum(n, x1, y1)); } } } return 0; }

 

【P.S】

 

上课去，回来后再继续奋斗！