84.柱状图中最大的矩形
题目链接:84.柱状图中最大的矩形
文档讲解:代码随想录
思路
与接雨水类似,不过此题需要找左右两侧第一个比自己小的元素。
当找到左右两侧第一个比自己小的元素后,当前元素高度的柱子的宽度就确定下来。
代码
class Solution {
public:
int largestRectangleArea(vector<int> &heights) {
int result = 0;
stack<int> st;
heights.insert(heights.begin(), 0); // 数组头部加入元素0
heights.push_back(0); // 数组尾部加入元素0
st.push(0);
for (int i = 1; i < heights.size(); i++) {
if (heights[i] > heights[st.top()])
st.push(i);
else if (heights[i] == heights[st.top()]) {
st.pop();
st.push(i);
} else {
while (!st.empty() && heights[i] < heights[st.top()]) {
int mid = st.top();
st.pop();
if (!st.empty()) {
int left = st.top();
int right = i;
int w = right - left - 1;
int h = heights[mid];
result = max(result, w * h);
}
}
st.push(i);
}
}
return result;
}
};