二维线段树(单点更新)
2015-08-19 16:45
302 查看
#include <iostream> #include <cstdio> #include <cstring> #include <cmath> #include <string> #include <vector> #include <queue> using namespace std; const int INF=1<<30; const int maxn=2000+10; struct INtervalTree2D{ //2维线段树x轴y轴的下标均从1开始 int Max[maxn][maxn],Min[maxn][maxn],Sum[maxn][maxn],n,m; int xo,xleaf,x1,y1,x2,y2,x,y,v,vmax,vmin,vsum; void query1D(int o,int L,int R){ if(y1<=L&&R<=y2){ vmax=max(Max[xo][o],vmax);vmin=min(Min[xo][o],vmin); vsum+=Sum[xo][o]; }else{ int M=L+(R-L)/2; if(y1<=M)query1D(o*2,L,M); if(M<y2)query1D(o*2+1,M+1,R); } } void query2D(int o,int L,int R){ if(x1<=L&&R<=x2){xo=o;query1D(1,1,m);} else{ int M=L+(R-L)/2; if(x1<=M)query2D(o*2,L,M); if(M<x2)query2D(o*2+1,M+1,R); } } void modify1D(int o,int L,int R){ if(L==R){ if(xleaf){Sum[xo][o]=Max[xo][o]=Min[xo][o]=v;return ;} Max[xo][o]=max(Max[xo*2][o],Max[xo*2+1][o]); Min[xo][o]=min(Min[xo*2][o],Min[xo*2+1][o]); Sum[xo][o]=Sum[xo*2][o]+Sum[xo*2+1][o]; }else { int M=L+(R-L)/2; if(y<=M){modify1D(o*2,L,M);} else {modify1D(o*2+1,M+1,R);} Max[xo][o]=max(Max[xo][o*2],Max[xo][o*2+1]); Min[xo][o]=min(Min[xo][o*2],Min[xo][o*2+1]); Sum[xo][o]=Sum[xo][o*2]+Sum[xo][o*2+1]; } } void modify2D(int o,int L,int R){ if(L==R){xo=o;xleaf=1;modify1D(1,1,m);} else{ int M=L+(R-L)/2; if(x<=M)modify2D(o*2,L,M); else modify2D(o*2+1,M+1,R); xo=o;xleaf=0;modify1D(1,1,m); } } void init_size(int h,int w)//设置矩阵的行数和列数 { n=h;m=w; } void modify(int x,int y,int v){ //将(x,y)的值修改为v(初始化线段树也是用该函数,调用m*n次) this->x=x;this->y=y;this->v=v; modify2D(1,1,n); } void query(int x1,int y1,int x2,int y2){ //询问x1,y1,x2,y2间的最值,调用该函数后,vmax,vmin即为结果 this->x1=x1;this->y1=y1;this->x2=x2;this->y2=y2; vmax=-INF;vmin=INF;vsum=0; query2D(1,1,n); } }xdtree; int main() { // freopen("in.txt","r",stdin); int n,m; int x,y,v; cin>>n>>m; xdtree.init_size(n,m); for(int i=1;i<=n;i++) for(int j=1;j<=m;j++) { scanf("%d",&v); xdtree.modify(i,j,v); } int q,x1,x2,y1,y2; cin>>q; char cas[3]; while(q--) { scanf("%s",cas); if(cas[0]=='q') { cin>>x1>>y1>>x2>>y2; xdtree.query(x1,y1,x2,y2); cout<<xdtree.vmax<<' '<<xdtree.vmin<<' '<<xdtree.vsum<<endl; } else { cin>>x>>y>>v; xdtree.modify(x,y,v); } } return 0; }
相关文章推荐
- Excel 用row()函数 在Excel中自动添加序号,
- LA 3135 Argus (优先队列的简单应用)
- Spring Quartz 框架结构概述
- OC中数组的四种遍历方式
- Big Christmas Tree
- Android 点击空白处隐藏键盘
- load-on-startup 解释
- struts note1
- ACM_阶段性总结 ACM_动态规划(DP)
- IntelliJ IDEA 工具的常见使用
- MySQL ODBC创建
- Redis在windows下的安装使用
- ZOJ3827——Information Entropy
- android开发工具
- 堆排序
- 求一个整数的所有位相加之和
- Android自定义布局
- apache的https访问配置
- ToggleButton
- 线程初体验之跑马灯