二维前缀和（C++）

幽影徘徊心之隅—冷光划破暗夜城

于 2024-10-07 15:10:11 发布

阅读量228

点赞数 19

分类专栏： C++课程讲解文章标签： c++ 开发语言

本文链接：https://blog.csdn.net/hjxxlsx/article/details/142740915

版权

C++课程讲解专栏收录该内容

15 篇文章 0 订阅

订阅专栏

一、二维前缀和的定义

二维前缀和是建立在一维前缀和之上的概念。在一个二维数组（矩阵）中，我们定义一个前缀和数组（通常用DP表示），其中DP[i][j]表示从矩阵左上角的点(1, 1)到点(i, j)这个子矩阵区域内所有元素的总和。例如，对于一个二维矩阵a[][]，我们可以通过一定的计算方法得到这个前缀和矩阵DP[][]。这就像是在二维平面上构建了一种累积的和的表示，每个点DP[i][j]的值都是以(1, 1)为起点到该点所构成的子矩阵内所有元素的和。这种定义方式为我们在处理与矩阵子区域求和相关的问题时提供了极大的便利，它将复杂的子矩阵求和问题转化为对已经预处理好的前缀和矩阵的简单计算，避免了每次查询子矩阵和时都要重新遍历子矩阵中的所有元素。就像在处理图像（图像可以看作是一个二维矩阵，每个像素点有对应的数值）时，如果我们需要频繁地获取图像中某个子区域的像素值总和，使用二维前缀和就可以大大提高计算效率。

二、二维前缀和的应用场景

（一）图像处理

在图像处理领域，图像可以被看作是一个二维矩阵，每个像素点具有一定的数值（例如灰度值等）。当我们需要对图像中的某个子区域进行分析时，比如计算子区域的平均像素值或者像素值总和等操作，二维前缀和就非常有用。例如，在进行图像的局部对比度增强时，可能需要先计算某个小区域内的像素值总和，通过二维前缀和，我们可以快速得到这个总和，而不需要对每个小区域都重新遍历其中的像素点。这样可以大大提高图像处理的速度，特别是在处理大型图像或者需要频繁计算子区域像素和的情况下。

（二）动态规划

在很多动态规划问题中，如果问题的模型可以转化为二维矩阵的形式，并且涉及到对矩阵子区域的某种计算（如求和等操作），二维前缀和就能够发挥作用。例如，在计算某个游戏棋盘（可以看作是二维矩阵）上某个区域的得分总和时，我们可以预先计算出二维前缀和，然后在不同的游戏状态下快速获取不同子区域的得分总和，从而为动态规划算法提供必要的数据支持，优化算法的时间复杂度。

（三）矩阵区域求和相关问题

当我们面对一个大型矩阵，并且需要多次查询这个矩阵中不同子矩阵的元素和时，二维前缀和能够在预处理之后以非常快的速度给出结果。比如在数据分析领域，有一个表示不同地区不同时间段数据的矩阵，我们可能需要查询某个地区在某个时间段内的数据总和（这个地区和时间段就对应着矩阵中的一个子矩阵），使用二维前缀和可以高效地解决这类问题，而避免每次查询都重新遍历子矩阵中的元素，提高了计算效率和查询速度。

三、二维前缀和的计算方法

基本思路
- 首先，我们要像计算一维前缀和那样，从左上角开始逐行逐列地计算二维前缀和。对于一个大小为n * m的二维矩阵a[][]，我们定义其前缀和矩阵为DP[][]。计算过程是基于递推的思想。
递推公式推导
- 对于矩阵中的第一行和第一列元素，计算比较特殊。当i = 1且j = 1时，DP[1][1]=a[1][1]；当i = 1且j>1时，DP[1][j]=DP[1][j - 1]+a[1][j]，这就相当于在第一行上按照一维前缀和的方式进行计算；同理，当i>1且j = 1时，DP[i][1]=DP[i - 1][1]+a[i][1]，这是在第一列上的计算。
- 对于矩阵中其他位置的元素（i>1且j>1），DP[i][j]=DP[i - 1][j]+DP[i][j - 1]-DP[i - 1][j - 1]+a[i][j]。这里的DP[i - 1][j]表示的是包含当前列上面部分的前缀和，DP[i][j - 1]表示的是包含当前行左边部分的前缀和，而DP[i - 1][j - 1]被重复计算了一次（它既包含在DP[i - 1][j]中又包含在DP[i][j - 1]中），所以要减去一次，最后再加上当前位置的元素a[i][j]。
计算示例
- 假设我们有一个3 * 3的矩阵a[][]如下：
  - \begin{bmatrix}1&2&3\\4&5&6\\7&8&9\end{bmatrix}147258369
- 首先计算第一行和第一列的前缀和：
  - DP[1][1]=a[1][1]=1；
  - DP[1][2]=DP[1][1]+a[1][2]=1 + 2 = 3；
  - DP[1][3]=DP[1][2]+a[1][3]=3+3 = 6；
  - DP[2][1]=DP[1][1]+a[2][1]=1+4 = 5；
- 然后计算其他位置的元素：
  - DP[2][2]=DP[1][2]+DP[2][1]-DP[1][1]+a[2][2]=3 + 5-1+5 = 12；
  - DP[2][3]=DP[1][3]+DP[2][2]-DP[1][2]+a[2][3]=6+12 - 3+6 = 21；
  - DP[3][2]=DP[2][2]+DP[3][1]-DP[2][1]+a[3][2]=12+12 - 5+8 = 27；
  - DP[3][3]=DP[2][3]+DP[3][2]-DP[2][2]+a[3][3]=21+27 - 12+9 = 45；
- 这样我们就得到了前缀和矩阵DP[][]：
  - \begin{bmatrix}1&3&6\\5&12&21\\12&27&45\end{bmatrix}15123122762145 。

四、二维前缀和的代码实现示例

以下是一个C++ 语言实现二维前缀和计算的示例代码：

#include <iostream>
#include <vector>
 
using namespace std; 
 
// 计算二维前缀和
vector<vector<int>> get2DPrefixSum(vector<vector<int>>& matrix) {
    int n = matrix.size();  
    int m = matrix[0].size(); 
    // 创建前缀和矩阵并初始化第一行和第一列
    vector<vector<int>> prefixSum(n + 1, vector<int>(m + 1, 0)); 
    for (int i = 1; i <= n; i++) {
        for (int j = 1; j <= m; j++) {
            // 根据递推公式计算前缀和
            prefixSum[i][j]=prefixSum[i - 1][j]+prefixSum[i][j - 1]-prefixSum[i - 1][j - 1]+matrix[i - 1][j - 1]; 
        }
    }
    return prefixSum; 
}
 
int main() {
    // 示例矩阵
    vector<vector<int>> matrix = {
        {1, 2, 3}, 
        {4, 5, 6}, 
        {7, 8, 9}
    }; 
    vector<vector<int>> prefixSum = get2DPrefixSum(matrix); 
    // 输出前缀和矩阵
    for (int i = 1; i <= matrix.size();  i++) {
        for (int j = 1; j <= matrix[0].size(); j++) {
            cout << prefixSum[i][j] << " "; 
        }
        cout << endl; 
    }
    return 0; 
}

在这段代码中：

首先定义了一个函数get2DPrefixSum，它接受一个二维矩阵matrix作为参数。
在函数内部，创建了一个比原矩阵行列数都大1的二维向量prefixSum，用于存储前缀和。
然后通过两层嵌套的for循环，按照二维前缀和的递推公式计算prefixSum矩阵的值。
在main函数中，创建了一个示例矩阵，调用get2DPrefixSum函数得到前缀和矩阵，并输出这个前缀和矩阵。这只是一个简单的示例，在实际应用中，可以根据具体的需求修改代码，比如查询特定子矩阵的和等操作。

五、二维前缀和常见问题及解决办法

（一）下标越界问题

问题描述
- 在计算二维前缀和或者使用已计算好的前缀和矩阵时，很容易出现下标越界的情况。例如，在计算前缀和矩阵时，如果没有正确处理边界情况（如第一行、第一列元素的计算），或者在查询子矩阵和时，传入的子矩阵坐标超出了前缀和矩阵的范围，就会导致下标越界错误。
解决办法
- 在编写代码时，要特别注意边界条件的处理。对于计算前缀和矩阵，确保在计算第一行和第一列元素时使用单独的计算逻辑，如前面提到的当i = 1且j = 1时，DP[1][1]=a[1][1]；当i = 1且j>1时，DP[1][j]=DP[1][j - 1]+a[1][j]；当i>1且j = 1时，DP[i][1]=DP[i - 1][1]+a[i][1]。在查询子矩阵和时，要对传入的坐标进行合法性检查，确保坐标在有效范围内。

（二）计算逻辑错误

问题描述
- 二维前缀和的计算逻辑相对复杂，尤其是对于递推公式的理解和应用。如果在计算过程中错误地理解或者应用了递推公式，就会导致计算出的前缀和矩阵错误。例如，可能会忘记减去重复计算的部分（在计算DP[i][j]（i>1且j>1）时忘记减去DP[i - 1][j - 1]），或者在计算过程中混淆了行列的关系。
解决办法
- 深入理解二维前缀和的计算原理，尤其是递推公式的推导过程。可以通过手动计算一些简单的示例矩阵的前缀和来加深理解，如前面计算3 * 3矩阵前缀和的示例。在编写代码时，可以添加一些注释来明确每一步计算的意义，方便自己检查和他人阅读代码。同时，进行充分的测试，使用不同规模和内容的矩阵进行测试，确保计算结果的正确性。