算法训练营day60_单调栈(3.25提前写)
84.柱状图中最大的矩形
对于每个柱子,找到左边第一个小于自己的柱子,以及右边第一个小于自己的柱子;
这个与接雨水恰好相反,这个有点像凸出来,而接雨水是凹进去;
维护一个单调递增的单调栈;
当前元素比栈顶元素小时,往前清算;
假设清算x,x左边第一个小于自己的柱子就是当前栈顶(已经把x弹出来了),x右边第一个小于自己的柱子是i;
注意,存在边界问题:
- 有可能最后一直递增,没有小元素来清算最后的,所以往后面加个0;
- 当清算到栈里最后一个元素时,他左边没有小于自己的,所以l是0号柱子,i-1-0+1还是i;
class Solution {
public:
int stk[100010],top;
int largestRectangleArea(vector<int>& heights) {
memset(stk,0,sizeof stk);
top=0;
heights.push_back(0);
int n=heights.size();
int ans=0;
for(int i=0;i<n;i++){
while(top&&heights[i]<heights[stk[top]]){
int t=stk[top];
top--;
if(!top){
ans=max(ans,i*heights[t]);
}
else{
int l=stk[top]+1;
ans=max(ans,(i-l)*heights[t]);
}
}
stk[++top]=i;
}
return ans;
}
};