leetcode 面试题 17.24. 最大子矩阵【二维前缀和+最长子序列】

给定一个正整数、负整数和 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

code:

def getMaxMatrix(matrix):
    max_ = float('-inf')
    m, n = len(matrix), len(matrix[0])  # m行，n列
    ans = [0]*4
    # 列的前缀和
    sum_col = [[0] * n]
    for i in range(1, m+1):
        sum_col.append([])
        for j in range(0, n):
            sum_col[i].append(sum_col[i-1][j] + matrix[i-1][j])
    # 最大子矩阵
    for i in range(1, m+1):  # 枚举上边界
        for j in range(i, m+1):  # 枚举下边界
            dp = 0
            for k in range(0, n):
                if dp <= 0:  # 独立门户，成新子矩阵
                    dp = sum_col[j][k] - sum_col[i-1][k]
                    c1 = k
                else:  # 增大子矩阵
                    dp = dp + sum_col[j][k] - sum_col[i-1][k]
                if dp > max_:  # 更新最大子矩阵和及其坐标
                    max_ = dp
                    ans[0] = i - 1
                    ans[1] = c1
                    ans[2] = j - 1
                    ans[3] = k
    return ans

print(getMaxMatrix([
   [-1,0],
   [0,-1]
]))

参考：https://leetcode-cn.com/problems/max-submatrix-lcci/solution/zhe-yao-cong-zui-da-zi-xu-he-shuo-qi-you-jian-dao-/