序言
总结最大矩阵的两道相似问题
题目一:lt84
给定 n 个非负整数,用来表示柱状图中各个柱子的高度。每个柱子彼此相邻,且宽度为 1 。求在该柱状图中,能够勾勒出来的矩形的最大面积
思路是:用stack存起所有的index,如果遇到比 stack[-1] 位置上高度 小的,即可算出stack[-1] 位置上的最大面积。然后pop掉 。
def largestRectangleArea(heights):
heights = [0] + heights + [0]
n= len(heights)
satck = []
ans = 0
for i in range(n):
while stack and heights[i] < heights[stack[-1]]:
cur = stack.pop()
ans = max(ans, heights[cur] * (i-stack[-1]-1))
stack.append(i)
return ans
题目二: lt85
- 最大矩形
给定一个仅包含 0 和 1 、大小为 rows x cols 的二维二进制矩阵,找出只包含 1 的最大矩形,并返回其面积
输入:matrix = [["1","0","1","0","0"],["1","0","1","1","1"],
["1","1","1","1","1"],["1","0","0","1","0"]]
输出:6
解释:最大矩形如上图所示
输入:matrix = []
输出:0
输入:matrix = [["1"]]
输出:1
这里的思路:
对于每一层,每一列都要算上一次最大矩阵,用上84的方法
比如 例1:
对于第一层,数组为 [1,0,1,0,0], 计算最大矩阵面积
对于第二层,数组为[2,0,2,1,1], 计算最大矩阵面积
对于第三层,数组为 [3,1,3,2,2], 计算最大矩阵面积
对于第四层,数组为[4,0,0,3,0], 计算最大矩阵面积
这里的要点是 当这一层出现0,则相对应的我数值则为0,如果连续出现1,则累加。
def maximalRectangle( matrix: List[List[str]]):
if not matrix: return 0
row = len(matrix)
col = len(matrix[0])
ans = 0
nums = [0]*col
for i in range(row):
for j in range(col):
if matrix[i][j] =='1': nums[j]+=1
else: nums[j] = 0
#创建好了nums
heights = [0] + nums + [0]
N = len(heights)
stack = []
for n in range(N):
while stack and heights[n] < heights[stack[-1]]:
cur = stack.pop()
ans = max(ans, heights[cur] * (n-1-stack[-1]))
stack.append(n)
return ans