算法基础集训（第11天，共106天）------＞带你彻底搞懂【前缀和】之【子矩阵的和】，很多高校的校赛都考过

本文链接：https://blog.csdn.net/mary19831/article/details/126333347

一：预处理

二维前缀和涉及到一个很重要的操作：预处理

顾名思义：在输入的同时，就对数据进行一些处理，使得后续对问题解决有更方便的作用

问题的起源就是对一个矩阵，如何在最快的时间内，求出它的某个子矩阵的和。我们首先把问题简化，对于一个n*m 的矩阵 A，如何快速求出

$gif.latex?%5Csum_%7Bi%3D0%7D%5E%7Ba%7D%5Csum_%7Bj%3D0%7D%5E%7Bb%7Da%5Bi%5D%5Bj%5D$

即求出图中的红色部分

求出上图红色部分的和我们需要下面的四个图来辅助解释

在求红色部分的和时，我们确保绿色部分、蓝色部分、黄色部分的和都已经求出来了，则可以通过以下公式直接求解：红色部分 = A【a】【b】 + 绿色部分 + 蓝色部分 − 黄色部分

假设sum【i】【j】表示从a【0】【0】到a【i】【j】这个区域的和，我们就容易得到如下递推式

sum【a】【b】=a【a】【b】+sum【a-1】【b】+sum【a】【b-1】−sum【a-1】【b-1】

二：预处理代码

    for (int i = 1; i <= n; i++)
        for (int j = 1; j <= m; j++) 
           {
            scanf("%d", &a[i][j]);
            s[i][j] = s[i][j - 1] + s[i - 1][j] - s[i - 1][j - 1] + a[i][j]; // 求前缀和
           }

注意：我们这里预处理是直接从a【1】【1】开始的，从a【0】【0】开始也行，但是由于出现下标是负数的情况，所以还需要提前声明一下

三：题目描述

四：万年无误代码模板

#include <iostream>

using namespace std;

const int N = 1010;

int n,m,q
int a[N][N], s[N][N];

int main() {
    scanf("%d%d%d",&n,&m,&q);

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

    return 0;
}