给定 n 个非负整数,用来表示柱状图中各个柱子的高度。每个柱子彼此相邻,且宽度为 1 。
求在该柱状图中,能够勾勒出来的矩形的最大面积。
1.直观的解法:中心扩展法,即以某一个矩形为整个矩形块的最终高度,然后向两端延伸
int largestRectangleArea(vector<int>& heights) {
int maxV=0;
for(int i=0;i<heights.size();i++){
int l=i;
int r=i;
while(l>=0&&heights[l]>=heights[i])
l--;
while(r<heights.size()&&heights[r]>=heights[i])
r++;
int temp=(r-l-1)*heights[i];
if(temp>maxV)
maxV=temp;
}
return maxV;
}
2.但是显然超时了,可以发现其中有很多东西重复了,所以接着的思路是动态规划,用一个数组记录前边元素的左右边界。
int largestRectangleArea(vector<int>& heights) {
int maxV=0;
vector<vector<int>> dp(heights.size(),vector<int>(2,0));
for(int i=0;i<heights.size();i++){
if(i==0){
int r=i;
while(r<heights.size()&&heights[r]>=heights[i])
r++;
dp[i][0]=-1;
dp[i][1]=r;
}
else{
if(heights[i]>heights[i-1]){
int l=i-1;
int r=i;
while(r<heights.size()&&heights[r]>=heights[i])
r++;
dp[i][0]=l;
dp[i][1]=r;
}
else if(heights[i]==heights[i-1]){
dp[i][0]=dp[i-1][0];
dp[i][1]=dp[i-1][1];
}
else{
int l=dp[i-1][0];
int r=dp[i-1][1];
while(l>=0&&heights[l]>=heights[i])
l--;
while(r<heights.size()&&heights[r]>=heights[i])
r++;
dp[i][0]=l;
dp[i][1]=r;
}
}
int temp=dp[i][1]-dp[i][0]-1;
temp*=heights[i];
maxV=max(maxV,temp);
}
return maxV;
}
3.上面的方法虽然ac,但是,时间和空间复杂度都有一点高,可不可以在o(n)的时间复杂度内完成?借鉴之前的雨水问题,可以通过单调栈来完成。
- 等价于求所有柱子为右边界的矩形的最大面积的最大面积
- 当前柱子为右边界的矩形的左边界,为最后一个小于当前柱子高度的下一个,有明显的单点关系,可以使用单调栈
int largestRectangleArea(vector<int>& heights) {
int maxV = 0;
stack<int> stk;
stk.push(0);
heights.insert(heights.begin(), 0); //哨兵,不用考虑栈空的问题了
heights.push_back(0); //哨兵,不用考虑遍历完后,栈还有元素的情况,or 可以自己复制一份数组
for (int i = 0; i<heights.size(); i++) {
while (heights[stk.top()]>heights[i]) {
int h = heights[stk.top()];
stk.pop(); //之所以要先pop一下,是为了获得对应位置左边的扩展部分
int w = i - stk.top()-1;
int area = h*w;
maxV = max(maxV, area);
}
stk.push(i);
}
return maxV;
}
相应的关于单调栈的问题有,42,739,496,316,901,402,581