题目链接
这次是求一个元素两边离他最近的小于它的元素位置,所以要用到从栈头到栈底元素大小单调递减的栈。
其实逻辑和接雨水一样,在Day51:503.下一个更大元素II、42. 接雨水写了,希望看了上一篇可以写出这道题哈哈哈。
不过这里有一个操作,就是在数组的收尾加0。
尾加0:当数组全部单调递减时,例如{1,2,3,4},所有元素走不到计算步骤。但是有0就不一样了,最后一个元素遍历完还有一个比较存在。
首加0:当数组全部单调递减,例如{9,8,7,6},第一个元素加入了,但是缺少了一个比较值来确定矩形的宽度是多少,这时首位0就可以帮忙,让代码的表达更精简便利。
class Solution {
public:
int largestRectangleArea(vector<int>& heights) {
heights.insert(heights.begin(),0);
heights.push_back(0);
stack<int> st;
int mid;
int result=0;
st.push(0);
for(int i=1;i<heights.size();i++){
while(!st.empty()&&heights[st.top()]>heights[i]){
mid=st.top();
st.pop();
int tmp=(i-st.top()-1)*heights[mid];
if(result<tmp)result=tmp;
}
st.push(i);
}
return result;
}
};
截止到现在,打卡顺利完成,接下来就是二刷代码随想录,一切任重而道远!加油加油。(等待图论部分的更新ing)