题目链接:牛客网
题目描述:
给出一个只包含0和1的二维矩阵,找出最大的全部元素都是1的长方形区域,返回该区域的面积
解题思路:
本题可以基于Leetcode #84 Largest Rectangle in Histogram去解。
这个博主的辅助栈法:Leetcode #84 Largest Rectangle in Histogram进行了详细的讲解。
以下图片来自博客截图。
实现技巧: 在直方图最后添加一个高度为0的虚拟矩形条,这样保证一次遍历之后栈里面的矩形都被正确处理过了,否则需要再重复一遍。
注意问题: 一定要进行非空判断,否则会产生段错误。
对于本题来说,可以将问题转换为求每一行直方图的最大面积问题。
将初始矩阵:(例子来源于牛客网only_唯)
1 0 1 0 0
1 0 1 1 1
1 1 1 1 1
1 0 0 1 0
按照行进行遍历,如果为1,则直方图高度加一,如果为0,则高度变为零。遍历后的每一行直方图高度转换为以下矩阵:
1 0 1 0 0
2 0 2 1 1
3 1 3 2 2
4 0 0 3 0
对于每一行的矩阵进行Leetcode #84 Largest Rectangle in Histogram的计算可以得到该行的直方图的最大面积,如果大于最大面积则更新,从而最终得到矩阵的最大长方形面积。
class Solution {
public:
int maximalRectangle(vector<vector<char> > &matrix) {
int row=matrix.size();
//判断是否为空
if(row <= 0 ) return 0;
int col=matrix[0].size();
vector<int> height(col+1);
int maxArea=0;
for(int i=0;i<row;i++){
stack<int> st;
//求更新后的直方图矩阵
for(int j=0;j<col;j++){
if(matrix[i][j]=='1'){
height[j]++;
}
else{
height[j]=0;
}
}
//在直方图最后添加一个高度为0的虚拟矩形条,这样保证一次遍历之后栈里面的矩形都被正确处理过了,否则需要再重复一遍
height[col]=0;
int n=height.size();
for(int j=0;j<n;j++){
//求右边界
if(st.empty() || height[st.top()] < height[j]){
st.push(j);
}else{
//计算之前所有矩形的面积,并将矩形合并
while(!st.empty()&&height[j]<=height[st.top()] ){
int h=height[st.top()];
st.pop();
int w=st.empty()?j:j-(st.top()+1);
maxArea = max(maxArea, h * w);
}
st.push(j);
}
}
}
return maxArea;
}
};
测试用例:
vector<vector<char> > matrix;
vector<char> m;
m.push_back('1');
m.push_back('0');
m.push_back('1');
m.push_back('0');
m.push_back('0');
matrix.push_back(m);
m.clear();
m.push_back('1');
m.push_back('0');
m.push_back('1');
m.push_back('1');
m.push_back('1');
matrix.push_back(m);
m.clear();
m.push_back('1');
m.push_back('1');
m.push_back('1');
m.push_back('1');
m.push_back('1');
matrix.push_back(m);
m.clear();
m.push_back('1');
m.push_back('0');
m.push_back('0');
m.push_back('1');
m.push_back('0');
matrix.push_back(m);