面试题 17.24. 最大子矩阵

1.题目描述

给定一个正整数、负整数和 0 组成的 N × M 矩阵，编写代码找出元素总和最大的子矩阵。
返回一个数组 [r1, c1, r2, c2]，其中 r1, c1 分别代表子矩阵左上角的行号和列号，r2, c2 分别代表右下角的行号和列号。若有多个满足条件的子矩阵，返回任意一个均可。
注意：本题相对书上原题稍作改动
示例：

说明：
1 <= matrix.length, matrix[0].length <= 200

2.思路

需要将二维转化为一维，对于矩阵的每一列，我们将其加在一起，成为了一维上的一个数，二维矩阵的和转化为了一维数组的和。

这样我们就将二维问题转化为了一维问题，现在另一个问题就是怎么把所有情况都遍历到呢？
我们以第i行为第一行，向下延申，设最后一行为第j行，我们就在i这么一个范围内，将二维问题转化为一维问题，再求解最大子序列和

我们将当前i~j行组成的矩阵的每一列的和存放在数组b中，其余的工作就是在求最大子序列和，并且保存其左上角和右下角

3.代码

class Solution {
public:
    vector<int> getMaxMatrix(vector<vector<int>>& matrix) {
        vector<int> res(4);
        if(matrix.empty()){
            return res;
        }
        int m = matrix.size();
        int n = matrix[0].size();

        int r1 = 0, c1 = 0;//矩阵左上角坐标
        

        int sum = 0;
        int maxSum = INT_MIN;

        for(int i = 0;i < m;++i){//子矩阵的上边
            vector<int> temp(n, 0);//二维转化为一维数组
            for(int j = i;j < m;++j){//子矩阵的下边
                sum = 0;
                for(int k = 0;k < n;++k){
                    temp[k] += matrix[j][k];
                
                    if(sum > 0){
                        sum += temp[k];
                    }
                    else{
                        sum = temp[k];
                        r1 = i;
                        c1 = k;
                    }
                    if(sum > maxSum){ //更新答案
                        maxSum = sum;
                        res[0] = r1;
                        res[1] = c1;
                        res[2] = j;
                        res[3] = k;
                    }
                }
            }
        }
        return res;
    }
};

4.复杂度分析

时间复杂度：O(mmn)
空间复杂度：O(n)