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