刷题打卡day60 第九章 单调栈 ● 84.柱状图中最大的矩形

84.柱状图中最大的矩形

/*
单调栈方法
栈顶到栈底：从大到小
找遍历到的当前的和左右第一个比他小的，三者凑成的矩形。
*/

class Solution {
public:

    int largestRectangleArea(vector<int>& heights) {
        stack<int> st;
        //数组头尾部加入元素0，让原本数组的头部也有上一个元素，让原来的尾部也有下一个元素
        heights.insert(heights.begin(), 0); // 数组头部加入元素0
        heights.push_back(0); // 数组尾部加入元素0

        //第一个入栈元素手动入0
        st.push(0);
        int result = 0;

        // 第一个元素已经入栈，从下标1开始
        for (int i = 1; i < heights.size(); i++) {
            // 注意heights[i] 是和heights[st.top()] 比较 ，st.top()是下标
            //情况1
            if (heights[i] > heights[st.top()]) {
                st.push(i);
            }
            //情况2
             else if (heights[i] == heights[st.top()]) {
                st.pop(); // 这个可以加，可以不加，效果一样，思路不同
                st.push(i);
            } 
            //情况3
            else {
                while (heights[i] < heights[st.top()]) { // 注意是while
                    int mid = st.top();
                    st.pop();
                    int left = st.top();
                    int right = i;
                    int w = right - left - 1;
                    int h = heights[mid];
                    result = max(result, w * h);
                }
                st.push(i);
            }
        }
        return result;
    }
};