二维线段树模板

来源:互联网 发布:淘宝助理6.0官方下载 编辑:程序博客网 时间:2024/06/06 09:42
void update_x(int rooty, int rootx, int L, int R, int x, int a) //tree[rooty][rootx]对应的矩阵x方向上范围是[L,R]{ tree[rooty][rootx] += a; if( L == R ) return; int mid = (L + R )/2; if( x <= mid ) update_x(rooty,( rootx << 1) + 1, L ,mid, x, a);else update_x(rooty,( rootx << 1) + 2, mid + 1,R, x, a);}void update_y(int rooty, int L, int R, int y, int x, int a) //tree[rooty][rootx]对应的矩阵y方向上范围是[L,R] { update_x( rooty,0, 1, S, x,a); if( L == R) return; int mid = (L + R )/2;if( y <= mid ) update_y( ( rooty << 1) + 1, L, mid,y, x, a); elseupdate_y( ( rooty << 1) + 2, mid+1, R, y, x, a); }int query_x(int rooty, int rootx, int L, int R, int x1, int x2) { if (L == x1 && R == x2)return tree[rooty][rootx]; int mid = (L + R) / 2; if (x2 <= mid)return query_x(rooty, (rootx << 1) + 1, L, mid, x1, x2); else if (x1 > mid)return query_x(rooty, (rootx << 1) + 2, mid + 1, R, x1, x2); elsereturn query_x(rooty, (rootx << 1) + 1, L, mid, x1, mid) + query_x(rooty, (rootx << 1) + 2, mid + 1, R, mid + 1, x2); }int query_y(int rooty, int L, int R, int y1, int y2, int x1, int x2){if (L == y1 && R == y2)return query_x(rooty, 0, 1, S, x1, x2);int mid = (L + R) / 2;if (y2 <= mid)return query_y((rooty << 1) + 1, L,mid, y1, y2, x1, x2);if (y1 > mid)return query_y((rooty << 1) + 2,mid + 1, R, y1, y2, x1, x2);elsereturn query_y((rooty << 1) + 1, L,mid, y1, mid, x1, x2) +query_y((rooty << 1) + 2,mid + 1, R, mid + 1, y2, x1, x2);}

0 0