leetcode 84. 柱状图中最大的矩形
题目描述:
给定 n 个非负整数,用来表示柱状图中各个柱子的高度。每个柱子彼此相邻,且宽度为 1 。
求在该柱状图中,能够勾勒出来的矩形的最大面积。

输入:heights = [2,1,5,6,2,3] 输出:10 解释:最大的矩形为图中红色区域,面积为 10
解题思路:
看题目本身,跟经典的接雨水有点类似。但不同点是接雨水的高度是由左右边界的最小值决定,这题的高度则是由每个柱子本身决定。
1. 暴力解法:自己能想到的解法。由当前柱子向左右两边延伸,只要满足高度大于当前柱子的,就能继续延伸,直到遇到高度小于当前柱子或到达边界。这实际上是以每个柱子为中心,向左右去寻找当前柱子的边界,将边界作为矩形宽度,柱子高度为矩形长度。最后取面积最大的矩形为最终结果。时间复杂度O(n^2),会超时。
2. 单调栈:大四算法课上单独讲过这类算法。实际是拿空间换时间,维护一个单调递增的栈,一定程度上是保证了左边界,再通过遍历的方式寻找右边界。始终维护一个单调递增的栈,遇到小于栈顶元素高度的柱子,则需要将栈顶元素出栈,也就意味着对于当前栈顶的元素来说,找到了其左右边界,左边界就为栈的下一个元素(若不存在则边界为-1),右边界为当前遍历到元素。当遍历结束后,需要清空栈,此时的右边界的确定的,为数组的长度,需要确定的是左边界,同样为栈的下一个元素(若不存在则边界为-1)。
代码:
解法一:
class Solution {
public:
int largestRectangleArea(vector<int>& heights) {
int n = heights.size();
int ans = 0;
for(int i=0; i<n; i++) {
int l = i;
int r = i;
while(l>=0 && heights[i] <= heights[l]) {
l--;
}
while(r<n && heights[i] <= heights[r]) {
r++;
}
if(r>=n)
r = n-1;
else if(r != i)
r = r-1;
if(l<0)
l=0;
else if(l != i)
l = l+1;
int area = heights[i]*(r-l+1);
ans = max(ans, area);
}
return ans;
}
};
解法二:
class Solution {
public:
int largestRectangleArea(vector<int>& heights) {
int n = heights.size();
stack<int> s;
int ans = 0;
for(int i=0; i<n; i++) {
while(!s.empty() && heights[s.top()] > heights[i]) {
int tmp = s.top();
s.pop();
int length = heights[tmp];
int weight = i;
if(!s.empty())
weight = i-s.top()-1;
ans = max(ans, length*weight);
}
s.push(i);
}
int length = 0;
int weight = n;
while(!s.empty()) {
int tmp = s.top();
s.pop();
length = heights[tmp];
weight = n;
if(!s.empty())
weight = n-s.top()-1;
ans = max(ans, length*weight);
}
return ans;
}
};
浙公网安备 33010602011771号