题目
给定 n 个非负整数,用来表示柱状图中各个柱子的高度。每个柱子彼此相邻,且宽度为 1 。
求在该柱状图中,能够勾勒出来的矩形的最大面积。
示例 1:
输入:heights = [2,1,5,6,2,3]
输出:10
解释:最大的矩形为图中红色区域,面积为 10
示例 2:
输入: heights = [2,4]
输出: 4
提示:
1 <= heights.length <=105
0 <= heights[i] <= 104
解题思路
使用单调栈来求解。首先,我们需要找到每个柱子左边第一个比它小的柱子和右边第一个比它小的柱子。然后,计算以当前柱子为高度的矩形的面积,并更新最大面积。最后返回最大面积即可。
代码实现
class Solution {
public:
int largestRectangleArea(vector<int>& heights) {
int n = heights.size();
vector<int> left(n), right(n);
stack<int> st;
for (int i = 0; i < n; i++) {
while (!st.empty() && heights[st.top()] >= heights[i]) {
st.pop();
}
if (st.empty()) {
left[i] = -1;
} else {
left[i] = st.top();
}
st.push(i);
}
while (!st.empty()) {
st.pop();
}
for (int i = n - 1; i >= 0; i--) {
while (!st.empty() && heights[st.top()] >= heights[i]) {
st.pop();
}
if (st.empty()) {
right[i] = n;
} else {
right[i] = st.top();
}
st.push(i);
}
int maxArea = 0;
for (int i = 0; i < n; i++) {
maxArea = max(maxArea, heights[i] * (right[i] - left[i] - 1));
}
return maxArea;
}
};