前缀和（二维）

原创于 2024-01-26 11:54:10 发布 · 545 阅读

7 ·

CC 4.0 BY-SA版权

文章标签：

#算法 #c++

基础算法专栏收录该内容

2 篇文章

订阅专栏

该代码片段展示了如何使用C++编程语言计算二维数组a的前缀和，并实现快速查询指定矩形区域内的和。通过输入矩阵尺寸和查询，程序计算并输出子矩阵的和。

#include<iostream>
using namespace std;

const int N=1010;
int n,m,q;
int a[N][N],s[N][N];

int main()
{
    cin>>n>>m>>q;
    for(int i=1;i<=n;i++)
        for(int j=1;j<=m;j++)
            cin>>a[i][j];

    for(int i=1;i<=n;i++)
        for(int j=1;j<=m;j++)
            s[i][j]=s[i-1][j]+s[i][j-1]-s[i-1][j-1]+a[i][j];//求前缀和

    while(q--)
    {
        int x1,y1,x2,y2;
        cin>>x1>>y1>>x2>>y2;
        cout<<s[x2][y2]-s[x1-1][y2]-s[x2][y1-1]+s[x1-1][y1-1]<<'\n';
    }
    return 0;
}