代码随想录算法训练营第五十五天|42. 接雨水 84.柱状图中最大的矩形

LeetCode 42. 接雨水

题目链接：42. 接雨水

踩坑：st.push(0)写成st.push(height[0])了

思路：本题向我们拓展了单调栈的性质，栈口元素左边下一个更大的元素就是栈中其下的第一个元素。这样我们就能在遍历的过程中不断求出以栈口元素为底的凹地。

代码：

class Solution {
public:
    int trap(vector<int>& height) {
        stack<int> st;
        int result = 0;

        st.push(0);
        for(int i = 1; i < height.size(); i++)
        {
            if(height[i] <= height[st.top()]) st.push(i);
            else
            {
                while(!st.empty() && height[i] > height[st.top()])
                {
                    int mid = height[st.top()];
                    st.pop();
                    if(!st.empty())
                    {
                        int h = min(height[i] - mid, height[st.top()] - mid);
                        int w = i - st.top() - 1;
                        result += (h*w);
                    }
                }
                st.push(i);
            }
        }
        return result;
    }
};

LeetCode 84.柱状图中最大的矩形

题目链接：84.柱状图中最大的矩形

踩坑：无

思路：与接雨水正好相反，接雨水是求栈口元素左右下一个更大的元素，本题是求栈口元素左右下一个更小的元素，左右两个元素的宽就是该元素可以画矩形的宽

代码：

class Solution {
public:
    int largestRectangleArea(vector<int>& heights) {
        stack<int> st;
        int result = 0;
        heights.insert(heights.begin(), 0);
        heights.push_back(0);

        st.push(0);
        for(int i = 1; i < heights.size(); i++)
        {
            if(heights[i] >= heights[st.top()]) st.push(i);
            else
            {
                while(!st.empty() && heights[i] < heights[st.top()])
                {
                    int mid = heights[st.top()];
                    st.pop();
                    if(!st.empty())
                    {
                        int w = i - st.top() - 1;
                        if(w*mid > result) result = w*mid;
                    }
                }
                st.push(i);
            }
        }
        return result;
    }
};