AcWing 796. 子矩阵的和

题目描述

---

分析：

二维前缀和与一维 前缀和思路相同，首先是如何构造前缀和？接着是如何利用前缀和求区间(块)的和？

我们先看第一个问题，这里通过图像给出解答：

前缀和求解时需要按行进行，同时和一维一样，为避免边界问题。$s[0][0]$ 需置为 $0$。

接下来看如何利用前缀和矩阵求原矩阵中子矩阵的和。

本题输入最多会达到一百万个，因此我们采用scanf进行读入。

---

代码(C++)

#include <iostream>

using namespace std;

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

int main()
{
    cin >> n >> m >> q;
    
    for (int i = 1; i <= n; i ++)
        for (int j = 1; j <= m; j ++)
            scanf("%d", &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;
        scanf("%d%d%d%d", &x1, &y1, &x2, &y2);
        
        printf("%d\n", s[x2][y2] - s[x1 - 1][y2] - s[x2][y1 - 1] + s[x1 - 1][y1 - 1]);
    }
}

---