前缀和一镜到底：秒懂一、二维前缀和的逻辑与实现方式

一、一维前缀和

1.1预处理前缀和数组

如果用之前暴力遍历的方式来完成这道题，首先输入数据时间复杂度位O（n），然后遍历q次每次都从开始到结尾一个一个数进行加减操作，那么时间复杂度就为O（n*q）。毫无疑问数据量大的情况下绝对超时。此时可以采用前缀和的方法来解决此问题。

前缀和数组的含义就是该数组第i位上的数等于原数组从第1位到第i位的数据之和。有了这个数组，在求下标到下标区间的和时我们就可以直接进行减法操作来完成。首先可以看到数组从1开始到n，所以在定数组arr和前缀和数组dp时都可以直接将数组大小定为n+1，即方便遍历也为构建前缀和数组提供基础条件。

所以通过前缀和数组的定义可以得到前缀和状态转移方程：dp[i]=dp[i-1]+arr[i];

1.2使用前缀和数组

前缀和数组构建完成后，就可以通过前缀和数组来直接计算某一区间的和了，当要计算left到right区间的和时，dp[right]中存放的时第一位到第right的值，而ldp[eft]中存放的是第一位到第left位存放的值，而直接拿dp[right]-dp[left]就会把原本arr[left]的值减掉，所以应该减left前一位的值，此时可以得到求第left到right位置值的方程：

dp[right]-dp[left-1];

此时也可以看出，如果下标从0开始，如果要求第0-right位，那么0-1=-1就会越界，如果从1开始将第0位的值设置为0，dp[right]-dp[0];那么计算时依旧不会越界，并且不会影响最终计算结果。

代码实现

#include <iostream>
using namespace std;
#include<vector>
int main() 
{
    //读入数据
    int n=0,m=0,q=0;//n行m列，输入q次（x1,y1）-(x2,y2)坐标的数据进行测试
    cin>>n>>m>>q;
    vector<vector<int>> arr(n+1,vector<int>(m+1));
    //构建原始矩阵数组
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=m;j++)
        {
            cin>>arr[i][j];
        }
    }
    //预处理前缀和矩阵
    vector<vector<long long>> dp(n+1,vector<long long>(m+1));//防止溢出
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=m;j++)
        {
        dp[i][j]=dp[i-1][j]+dp[i][j-1]-dp[i-1][j-1]+arr[i][j];
        }
    }
    //计算前缀和
    int x1=0,y1=0,x2=0,y2=0;
    while(q--)
    {
        cin>>x1>>y1>>x2>>y2;
        cout<<dp[x2][y2]-dp[x1-1][y2]-dp[x2][y1-1]+dp[x1-1][y1-1]<<endl;
    }
    
    return 0;
}

二、二维前缀和

描述

给你一个 n 行 m 列的矩阵 A ，下标从1开始。

接下来有 q 次查询，每次查询输入 4 个参数 x1 , y1 , x2 , y2

请输出以 (x1, y1) 为左上角 , (x2,y2) 为右下角的子矩阵的和，

输入描述：

第一行包含三个整数n,m,q.

接下来n行，每行m个整数，代表矩阵的元素

接下来q行，每行4个整数x1, y1, x2, y2，分别代表这次查询的参数

2.1预处理出前缀和矩阵

首先和一维前缀和一样，进行数据的输入后先根据原始数据预处理出一个前缀和矩阵，方便接下来直接使用。

相比于一维前缀和，二维前缀和矩阵的构建就没有那么容易了，在二维前缀和矩阵中，dp[i] [j]表示的是从左上角坐标(1,1)到右下角坐标(i,j)的矩阵中所有元素的和，通过上图可以看到，要求右下标为(i,j)的矩阵的和，可以把矩阵划分成为四部分，这里区域标记为ABCD，从而求矩阵的和就可以变成对ABCD四部分的求和操作，d就是arr数组中（i，j）位置 存放的值，A+B就是dp中（i，j）上面存放的值，A+C就是（i，j）左边存放的值从而可以得到dp[i][j]=(A+B)+(A+C)+D-A;从而我们就可以列出二维前缀和矩阵的状态转移方程。

dp[i][j]= dp[i-1][j] + dp[i][j-1] + arr[i][j] - dp[i-1][j-1];

列出状态转移方程的这个过程时间复杂度为O(n*m).

2.2使用前缀和矩阵

前缀和矩阵构建完毕后，就可以根据矩阵来求特定矩阵的和了，和构建矩阵时一样，也可以将矩阵划分为四个区域ABCD，假设D就是我们要求的区域，思路和之前一样，可以列出公式D=A+C+B+D-(A+B)-(A+C)+A;从而可以得到求[x1,y1]到[x2,y2]矩阵的和的式子：

D=dp[x2,y2]-dp[x1-1,y2]-dp[x2,y1-1]+dp[x1-1][y1-1]; 

此时进行q此查询操作可以直接得出所求矩阵的和，所以前缀和算法的时间复杂度为O(n*m+q);

代码实现

#include <iostream>
using namespace std;
#include<vector>
int main() 
{
    //读入数据
    int n=0,m=0,q=0;//n行m列，输入q次（x1,y1）-(x2,y2)坐标的数据进行测试
    cin>>n>>m>>q;
    vector<vector<int>> arr(n+1,vector<int>(m+1));
    //构建原始矩阵数组
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=m;j++)
        {
            cin>>arr[i][j];
        }
    }
    //预处理前缀和矩阵
    vector<vector<long long>> dp(n+1,vector<long long>(m+1));//防止溢出
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=m;j++)
        {
        dp[i][j]=dp[i-1][j]+dp[i][j-1]-dp[i-1][j-1]+arr[i][j];
        }
    }
    //计算前缀和
    int x1=0,y1=0,x2=0,y2=0;
    while(q--)
    {
        cin>>x1>>y1>>x2>>y2;
        cout<<dp[x2][y2]-dp[x1-1][y2]-dp[x2][y1-1]+dp[x1-1][y1-1]<<endl;
    }
    
    return 0;
}