【代码随想录刷题记录】 84.柱状图中最大的矩形

84.柱状图中最大的矩形

1、题目
给定 n 个非负整数，用来表示柱状图中各个柱子的高度。每个柱子彼此相邻，且宽度为 1 。

求在该柱状图中，能够勾勒出来的矩形的最大面积。

题目链接：https://leetcode.cn/problems/largest-rectangle-in-histogram/

2、代码

class Solution {
public:
    int largestRectangleArea(vector<int>& heights) {
        if(heights.size() == 0) return 0;
        int result = 0;
        //确保每个元素都可以走完while循环中的内容
        heights.insert(heights.begin(), 0); // 数组头部加入元素0
        heights.push_back(0); // 数组尾部加入元素0
        stack<int> st;
        st.push(0);
        result = heights[0];
        for(int i = 1; i < heights.size(); i++){
            while(!st.empty() && heights[i] < heights[st.top()]){
                int mid = st.top();
                st.pop();
                if(!st.empty()){
                    int h = heights[mid];
                    int w = i - st.top() - 1;
                    int s = h * w;
                    result = max(result, s);
                }
            }
            st.push(i);
        }
        return result;


    }
};

3、小结
与接雨水类似，只不过本题需要在头尾加入0，使得从以所有柱子为中间的最大面积都可以被算出来。最后在遍历的过程中取最大值即可。