柱狀圖中最大的矩形
題目連結:84. 柱狀圖中最大的矩形 - 力扣(LeetCode)
思路:掌握了……嗎?還是參考了下官網思路。程式碼隨想錄 (programmercarl.com)
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;
result=max(result,w*heights[mid]);
}
}
st.push(i);
}
}
return result;
}
};