每日一道Leetcode - 面试题 17.24. 最大子矩阵【动态规划】

最新推荐文章于 2021-09-11 19:14:17 发布

奋进的杰西

最新推荐文章于 2021-09-11 19:14:17 发布

阅读量156

点赞数

分类专栏： Leetcode

本文链接：https://blog.csdn.net/weixin_41041275/article/details/115417603

版权

矩阵运算动态规划最大子矩阵 Python算法二维数组

关键词由CSDN通过智能技术生成

Leetcode 专栏收录该内容

304 篇文章 1 订阅

订阅专栏

在这里插入图片描述

class Solution:
    def getMaxMatrix(self, matrix: List[List[int]]) -> List[int]:
        N = len(matrix)
        M = len(matrix[0])
        # 初始化一维数组
        b = [0] * M
        max_res = float("-inf")
        ans = [0]*4
        for i in range(N):
            # 每行向下扩展遍历的时候先要将b数组初始化为0
            for k in range(M):
                b[k] = 0
            # 扩展到j行，一次将每列元素相加
            for j in range(i,N):
                cur_sum = 0
                for k in range(M):
                    b[k] += matrix[j][k]
                    # 看当前行遍历的每列的值是否最大，最大就是当前情况的最大子矩阵
                    if cur_sum <= 0:
                        # 前面是负数了，所以要另立门户
                        cur_sum = b[k]
                        c1 = k
                    else:
                        cur_sum += b[k]
                    # 找到比之前大的子矩阵
                    if cur_sum > max_res:
                        max_res = cur_sum
                        ans[0] = i
                        ans[1] = c1
                        ans[2] = j
                        ans[3] = k

        return ans