[YM]模板-二维前缀和模板及详解

夜喵YM

于 2024-08-10 23:09:58 发布

阅读量345

点赞数 6

文章标签：算法数据结构

本文链接：https://blog.csdn.net/2302_80481841/article/details/141097926

版权

前置：

看到这的你估计已经熟练了一维前缀和，那么我们向二维空间扩展一下

a[x][y]表示当前(x,y)的值

pre[x][y]表示(x,y)左上部分的前缀和

要求计算(x1,y1)到(x2,y2)这一部分的前缀和

公式：

pre[x2, y2] - pre[x1 - 1, y2] - pre[x2, y1 - 1] + pre[x1 - 1, y1 - 1]

模板：

不墨迹先上模板再证明

const int M = 1005;
int n,m,q;
ll a[M][M],pre[M][M];
void init(int n,int m){
    for(int x=1;x<=n;x++)
        for(int y=1;y<=m;y++){
            cin>>a[x][y];
            pre[x][y]=a[x][y]+pre[x-1][y]+pre[x][y-1]-pre[x-1][y-1];
        }
}
ll query(int Lx,int Rx,int Ly,int Ry){  
    return pre[Ly][Ry]-pre[Lx-1][Ry]-pre[Ly][Rx-1]+pre[Lx-1][Rx-1];
} 
void solve(){
    cin>>n>>m>>q;
    init(n,m);
    while(q--){
        int Lx,Rx,Ly,Ry;
        cin>>Lx>>Rx>>Ly>>Ry;
        cout<<query(Lx,Rx,Ly,Ry)<<"\n";
    }
}