首页 > 代码库 > POJ2155 Matrix 二维树状数组应用
POJ2155 Matrix 二维树状数组应用
一个N*N(1<=N<=1000)的矩阵,从(1,1)开始,给定一些操作C和一些查询Q,一共K条(1<=K<=50000):
C x1,y1,x2,y2 表示从x1行y1列到x2行y2列的元素全部反转(0变成1,1变成0);
Q x y表示询问x行y列的元素是0还是1。
题目乍一看感觉还是很难,如果能记录每一个元素的状态值,那答案是显而易见的,但是元素过多,如果每次都对每一个元素进行更新状态的话,复杂度太高。实际上只要记录边界的特定坐标的反转次数,最好的选择那就是二维树状数组了。
对于操作C,为了保证元素在子矩阵内的反转次数都增加了1,需要在Bit(x1,y1)的位置上加上1,Bit(x2 + 1, y1) 减去1,Bit(x1, y2 + 1)减去1,Bit(x2 +1, y2 + 1)加上1;
对于操作Q,直接获取Bit(x,y) 的值就代表该元素反转了多少次。
#include <stdlib.h> #include <stdio.h> #include <vector> #include <math.h> #include <string.h> #include <string> #include <iostream> #define MAXN 1001 template <class TYPE> void BITAdd(TYPE array[MAXN], int i, TYPE addvalue, int n) { while (i <= n) { array[i] += addvalue; i += i & -i; } } template <class TYPE> TYPE BITGet(TYPE array[MAXN], int i) { TYPE ss = 0; while (i > 0) { ss += array[i]; i -= i & -i; } return ss; } template <class TYPE> void BIT2DAdd(TYPE array[MAXN][MAXN], int i, int j,TYPE addvalue, int n, int m) { while (i <= n) { BITAdd<TYPE>(array[i], j, addvalue, m); i += i & -i; } } template <class TYPE> TYPE BIT2DGet(TYPE array[MAXN][MAXN], int i, int j) { TYPE ss = 0; while (i > 0) { ss += BITGet<TYPE>(array[i], j); i -= i & -i; } return ss; } typedef int MYTYPE; MYTYPE Bit[MAXN][MAXN]; int main() { #ifdef _DEBUG freopen("d:\\in.txt", "r", stdin); #endif int x; scanf("%d", &x); for (int i = 0; i < x; i++) { memset(Bit, 0, sizeof(Bit)); int n, k; scanf("%d %d\n", &n, &k); if (i >= 1) { printf("\n"); } for (int j = 0; j < k; j++) { char ope; scanf("%c", &ope); if (ope == 'C') { int x1, x2, y1, y2; scanf("%d %d %d %d\n", &x1, &y1, &x2, &y2); BIT2DAdd<int>(Bit, x1, y1, 1, n, n); BIT2DAdd<int>(Bit, x2 + 1, y1, -1, n, n); BIT2DAdd<int>(Bit, x1, y2 + 1, -1, n, n); BIT2DAdd<int>(Bit, x2 + 1, y2 + 1, 1, n, n); } else { int x, y; scanf("%d %d\n", &x, &y); int thisvalue = http://www.mamicode.com/BIT2DGet(Bit, x, y);> POJ2155 Matrix 二维树状数组应用
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。