因为leetCode 的85题是84题的变种,所以合成一个博客写。
84题的题意:一个直方图中,每个矩阵都是宽为1,高度不同。求在这个直方图里能画出矩形的最大面积
思路:先从前到后,再从后到前,重复如下过程:
用一个栈,先入第一个元素,,如果后一个元素的高度比栈顶低,则加上该数对应的宽度并弹出该元素。一直重复完。
代码如下:
int largestRectangleArea(vector<int>& heights)
{
stack<int> r;
int i,j=0,k,size=heights.size(),le=size,ans=0;
if(size==0) return 0;
vector<int> left(heights.size(),0),right(heights.size(),0);
for(i=0; i<size; i++)
{
while(!r.empty()&&heights[i]<=heights[r.top()])
{
left[i]+=1+left[r.top()];
r.pop();
}
r.push(i);
}
while(!r.empty()) r.pop();
for(i=size-1; i>=0; i--)
{
while(!r.empty()&&heights[i]<=heights[r.top()])
{
right[i]+=1+right[r.top()];
r.pop();
}
r.push(i);
}
for(i=0; i<size; i++) ans=max(ans,(left[i]+right[i]+1)*heights[i]);
return ans;
}
85题:有个m*n的矩阵,每个矩阵是0或1,求全1矩形面积的最大值。
很简单,每一行就是一个直方图的底,连续多少个1就是多少高度。与84题的代码一样,只不过重复。
代码如下:
int largestRectangleArea(vector<int>& heights)
{
stack<int> r;
int i,j=0,k,size=heights.size(),le=size,ans=0;
if(size==0) return 0;
vector<int> left(heights.size(),0),right(heights.size(),0);
for(i=0; i<size; i++)
{
while(!r.empty()&&heights[i]<=heights[r.top()])
{
left[i]+=1+left[r.top()];
r.pop();
}
r.push(i);
}
while(!r.empty()) r.pop();
for(i=size-1; i>=0; i--)
{
while(!r.empty()&&heights[i]<=heights[r.top()])
{
right[i]+=1+right[r.top()];
r.pop();
}
r.push(i);
}
for(i=0; i<size; i++) ans=max(ans,(left[i]+right[i]+1)*heights[i]);
return ans;
}
int maximalRectangle(vector<vector<char>>& matrix) {
int i,j,k,ans=0,m,n;
m=matrix.size();
if(m==0) return 0;
n=matrix[0].size();
vector<int> old(n,0);
for(i=0;i<m;i++)
{
for(j=0;j<n;j++)
{
if(matrix[i][j]=='1') old[j]++;
else old[j]=0;
}
ans=max(ans,largestRectangleArea(old));
}
return ans;
}