题目地址(17.24. 最大子矩阵)

题目地址(17.24. 最大子矩阵)

https://leetcode.cn/problems/max-submatrix-lcci/

题目描述

给定一个正整数、负整数和 0 组成的 N × M 矩阵，编写代码找出元素总和最大的子矩阵。

返回一个数组 [r1, c1, r2, c2]，其中 r1, c1 分别代表子矩阵左上角的行号和列号，r2, c2 分别代表右下角的行号和列号。若有多个满足条件的子矩阵，返回任意一个均可。

注意：本题相对书上原题稍作改动

示例：

输入：
[
   [-1,0],
   [0,-1]
]
输出：[0,1,0,1]
解释：输入中标粗的元素即为输出所表示的矩阵

 

说明：

1 <= matrix.length, matrix[0].length <= 200

前置知识

公司

• 暂无

思路

关键点

代码

• 语言支持：Python3

Python3 Code:

class Solution:
    def getMaxMatrix(self, matrix: List[List[int]]) -> List[int]:
        m, n = len(matrix), len(matrix[0])
        ans = [0] * 4
        r1, c1, r2, c2 = [0] * 4
        max_sum = matrix[0][0] # 全局最大和
        for i in range(m): # 起始行
            total = [0] * n
            for j in range(i, m): # 结束行
                cur_max = 0
                for k in range(n):
                    # 其实如果不需要统计坐标的只是寻找最大和的话
                    """
                    根连续的子序列的最大和的题写法是一样的
                    cur_max = max(cur_max, cur_max + total[k])
                    max_sum = max(cur_max, max_sum)
                    
                    只不过这里要求统计左上角和右下角的坐标，所以要使用if else 拆开然后记录行号和列号
                    """
                    total[k] += matrix[j][k]
                    if cur_max > 0: # cur_max + total[k] > cur_max 加了比不加更大
                        cur_max += total[k] # 那就加上去
                    else: # cur_max + total[k] <= cur_max 加了变得更小了
                        cur_max = total[k] # 那就不用加了
                        r1 = i # 更新行起点
                        c1 = k # 更新列起点
                    if cur_max > max_sum: # 如果当前和比全局最大和大了
                        r2, c2 = j, k # 更新一下行终点和列终点
                        max_sum = cur_max # 更新当前和为全局最大和
                        ans = r1, c1, r2, c2 # 更新结果
        return ans

复杂度分析

令 n 为数组长度。

• 时间复杂度：$O(n^3)$
• 空间复杂度：$O(n^3)$