给定一个由 0 和 1 组成的矩阵 matrix ,找出只包含 1 的最大矩形,并返回其面积。
注意:此题 matrix 输入格式为一维 01 字符串数组。
示例1:
输入:matrix = [“10100”,“10111”,“11111”,“10010”]
输出:6
解释:最大矩形如上图所示。
示例 2:
输入:matrix = []
输出:0
示例 3:
输入:matrix = [“0”]
输出:0
示例 4:
输入:matrix = [“1”]
输出:1
示例 5:
输入:matrix = [“00”]
输出:0
提示:
- rows == matrix.length
- cols == matrix[0].length
- 0 <= row, cols <= 200
- matrix[i][j] 为 ‘0’ 或 ‘1’
思路:很巧妙的思维:把当前行能组成的最大矩形,看作以当前行为底所表示的直方图的最大矩形。上一题求出了直方图的最大矩形面积,这里只需要一次把每一行转换成以当前行为底的直方图,调用求直方图最大矩形面积的函数。
int maximalRectangle(vector<string>& matrix) {
if (matrix.size() == 0) {
return 0;
}
vector<int> heights(matrix[0].size(), 0);
int maxArea = 0;
for (int i = 0; i < matrix.size(); ++i) {
for (int j = 0; j < matrix[0].size(); ++j) {
if (matrix[i][j] == '0') {
heights[j] = 0;
}
else {
heights[j] ++;
}
}
maxArea = max(maxArea, largestRectangleArea(heights));
}
return maxArea;
}
int largestRectangleArea(vector<int>& heights) {
stack<int> sta;
sta.push(-1);
int maxArea = 0;
heights.push_back(0);
for (int i = 0; i < heights.size(); ++i) {
while (sta.top() != -1 && heights[sta.top()] >= heights[i]) {
int height = heights[sta.top()];
sta.pop();
int width = i - sta.top() - 1;
maxArea = max(maxArea, height * width);
}
sta.push(i);
}
return maxArea;
}