二维线段树(单点更新)

#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;
}

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值