【Maximal Rectangle】cpp
题目:
Given a 2D binary matrix filled with 0's and 1's, find the largest rectangle containing all ones and return its area.
代码:
class Solution { public: int maximalRectangle(vector<vector<char> >& matrix) { if (matrix.empty()) return 0; const int ROW = matrix.size(); const int COL = matrix[0].size(); vector<int> height(COL,0); vector<int> l_most(COL,0); vector<int> r_most(COL,COL-1); int max_area = 0; for ( int i = 0; i < ROW; ++i ) { int l_curr = 0; int r_curr = COL-1; // calculate current row's height for ( int j = 0; j < COL; ++j ) { height[j] = matrix[i][j]=='1' ? height[j]+1 : 0; } // from left to right for ( int j = 0; j < COL; ++j ) { if ( matrix[i][j]=='1' ) { l_most[j] = std::max(l_most[j], l_curr); } else { l_most[j] = 0; l_curr = j+1; } } // from right to left for ( int j = COL-1; j>=0; --j ) { if ( matrix[i][j]=='1' ) { r_most[j] = std::min(r_most[j], r_curr); } else { r_most[j] = COL-1; r_curr = j-1; } } // calculate area of rectangle for ( int j = 0; j<COL; ++j ) { max_area = std::max(max_area, (r_most[j]-l_most[j]+1)*height[j]); } } return max_area; } };
tips:
学习了网上的O(m×n)时间复杂度的解法。
这道题的路子类似于http://www.cnblogs.com/xbf9xbf/p/4499450.html
总体的思路如下:
每次遍历一行的元素,以这一行作为X轴;模仿largest rectangle in histogram这道题
大体思路如下,需要对largest rectangle in histogram这道题有比较深刻的理解
a) 计算每个点的当前高度
b) 计算当前高度下每个点能往左推到哪个位置
c) 计算当前高度下每个点能往右推到哪个位置
d) 遍历以当前行作为X轴可能获得的最大面积
a)~d)走完所有行,就可以获得最大的矩形面积
思路实在精妙,详细记录下思考每个环节的过程:
a) 如何计算当前点的高度?
利用滚动数组技巧:height[j]记录的是到i-1行该列对应的高度;如果matrix[i][j]=='1',则height[j] += 1,即比上一行该列位置+1;如果matrix[i][j]=='0'则对于第i行来说,该列对应的高度就是0。这个道理比较直观,简单说就是滚动数组height[j]的更新原则是看高度能不能在新的一行续上。这种滚动数组技巧的好处在于只需要维护一个一维数组,否则需要维护二维数组,是dp过程的常用技巧。
b) c) 如何找每个点向左(右)能推到的最远的位置?
这个部分更是精妙,不仅利用了滚动数组的技巧,而且还充分利用了上一轮dp的结果。以b)为例分析,c)同理。
在此题的背景下,当前行的某列能向左推到哪要考虑如下几种情况:
1)高度是是0(即matrix[i][j]=='1'不成立):显然能推到最左侧即0的位置,不受上一行该列能向左推到哪的影响。
2)如果高度不是0(即matrix[i][j]=='1'成立),则这当前行该列能向左推到哪取决于左侧第一个不比它高的列在哪?
考虑一个问题:在matrix[i][j]=='1'的前提下,第i行的l_most[j]可不可能比第i-1行的l_most[j]还小(即还往左)?
显然,这是不可能的。因为第i行的j列已经把高度续上了,当且仅当第i行l_most[j]~j列的位置都满足高度续上的前提下,第i行的l_most[j]才可能等于第i-1行的l_most[j];一旦第i行l_most[j]~j有一个位置没有续上高度,那么第i行的l_most[j]就要比第i-1行的l_most[j]小了.
通过以上分析,在第i行j列已经把高度续上的前提下,能向左推到哪,关键取决于在j左边且最靠近j列没续上高度的列,是否影响到了l_most[j]~j。因此,维护一个l_curr,标记到当前列为止,续上高度的列最多可能向左推到哪。
所以,需要比较l_most[j]与l_curr的位置。如果遇上了高度为0的,则自动把l_curr+1;如果高度不为0的,判断l_curr是否影响到了l_most[j]
描述的比较绕口,但是客观上也就是这样了。
d) 计算可能的最大高度
这个就是个常规的dp,不再赘述了。注意一点就是(right-left+1)*height,这里是否“+1”取决于l_most和r_most的取值方式。
==============================================
第二次过这道题,思路记得比较清晰;具体操作上一些细节再熟练一下。
class Solution { public: int maximalRectangle(vector<vector<char> >& matrix) { int ret = 0; if ( matrix.empty() ) return ret; vector<int> height(matrix[0].size(), 0); for ( int i=0; i<matrix.size(); ++i ) { // update height for ( int j=0; j<matrix[i].size(); ++j ) { height[j] = matrix[i][j]=='0' ? 0 : height[j]+1; } // update max area ret = max(ret, Solution::maxArea(height)); } return ret; } static int maxArea(vector<int>& height) { int ret = 0; height.push_back(0); stack<int> sta; for ( int i=0; i<height.size(); ++i ) { if ( sta.empty() || height[i]>height[sta.top()] ) { sta.push(i); continue; } while ( !sta.empty() && height[sta.top()]>=height[i] ) { int tmp = sta.top(); sta.pop(); if ( sta.empty() ) { ret = max( ret, i*height[tmp] ); } else { ret = max(ret, (i-sta.top()-1)*height[tmp] ); } } sta.push(i); } height.pop_back(); return ret; } };