Leetcode: 84 柱状图中最大的矩形
本题与昨天接雨水的问题正好相反,要寻找的是左右小于它的柱子。栈里从大到小的顺序,才能保证栈顶元素找到左右两边第一个小于栈顶元素的柱子。
这样总体逻辑是一样的,依旧考虑的是要输入元素与栈顶元素比较,大于小于等于的情况,在这里我们要找到左右小于的柱子,因此是在小于的情况下考虑。
最后的结果就是底乘以柱子的高度。
class Solution {
public:
int largestRectangleArea(vector<int>& heights) {
heights.insert(heights.begin(),0);
heights.push_back(0);//初始处理,确保两端一定能找到比栈顶元素小的元素
stack<int> st;
st.push(0);
int result = 0;
for(int i =1; i < heights.size(); i++){
while(!st.empty() && heights[i] < heights[st.top()]){//注意是while循环找每一个
int mid = st.top();//当前柱子的高度坐标
st.pop();
if(!st.empty()){
int w = i - st.top() - 1;//计算宽度
int h = heights[mid];//计算高度
result = max(result, w*h);
}
}
st.push(i);//元素入栈
}
return result;
}
};
也可以使用双指针的方法来保留每个元素左右小于该柱子的坐标。