程式碼隨想錄演算法訓練營第六十天|● 84.柱狀圖中最大的矩形

SandaiYoung發表於2024-03-28

柱狀圖中最大的矩形

題目連結: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;
    }
};

🥳🥳🥳結束啦😁😁😁

相關文章