【ACWing】796. 子矩阵的和

题目地址：

https://www.acwing.com/problem/content/798/

给定一个$n$行$m$列的整数矩阵$A$，和$q$次询问，每次询问包含四个整数$x_1,y_1,x_2,y_2$，表示一个子矩阵左上角和右下角的坐标。求这个子矩阵的和。坐标是从$1$开始计数的。

输入格式：
第一行包含三个整数$n$，$m$，$q$。接下来$n$行，每行包含$m$个整数，表示整数矩阵。接下来$q$行，每行包含四个整数$x_1,y_1,x_2,y_2$，表示一组询问。

输出格式：
共$q$行，每行输出一个询问的结果。

数据范围：
$1\le n,m\le 1000$
$1\le q\le 200000$
$1\le x_1\le x_2\le n$
$1\le y_1\le y_2\le m$
$-1000\le A[i][j]\le 1000$

思路是用前缀和。先求出前缀和矩阵$s$，递推公式为：$$s[i+1][j+1]=A[i][j]+s[i+1][j]+s[i][j+1]-s[i][j]$$针对每次询问，子矩阵和为：$$s[x_2][y_2]-s[x_2][y_1-1]-s[x_1-1][y_2]+s[x_1-1][y_1-1]$$代码如下：

#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 = 0; i < n; i++)
        for (int j = 0; j < m; j++)
            scanf("%d", &a[i][j]);

	// 求前缀和矩阵
    for (int i = 0; i < n; i++)
        for (int j = 0; j < m; j++) 
            s[i + 1][j + 1] = a[i][j] + s[i][j + 1] + s[i + 1][j] - s[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[x2][y1 - 1] - s[x1 - 1][y2] + s[x1 - 1][y1 - 1]);
    }

    return 0;
}

预处理时间复杂度$O(nm)$，每次询问时间$O(1)$，空间$O(nm)$。