Day.42 | 42.接雨水 84.柱状图中最大的矩形

42.接雨水

要点：右边第一个更大的元素，与当前元素，与当前元素的左边栈顶元素，构成一个凹槽，计算凹槽的面积，获得结果。计算结果的时候，当前元素会被pop掉，所以其实是一行一行的计算能接雨水的量。

class Solution {
public:
    int trap(vector<int>& height) {
        if (height.size() <= 2)
            return 0;
        stack<int> st;
        st.push(0);
        int sum = 0;
        for (int i = 1; i < height.size(); ++i) {
            if (height[i] < height[st.top()]) {
                st.push(i);
            } else if (height[i] == height[st.top()]) {
                st.pop();
                st.push(i);
            } else {
                while (!st.empty() && height[i] > height[st.top()]) {
                    int mid = st.top();
                    st.pop();
                    if (!st.empty()) {
                        int h = min(height[i], height[st.top()]) - height[mid];
                        int w = i - st.top() - 1;
                        sum += h * w;
                    }
                }
                st.push(i);
            }
        }
        return sum;
    }
};

84.柱状图中最大的矩形

要点：右边第一个更小的元素，与当前元素，与当前元素左边的栈顶元素，构成一个凸形，计算凸的面积，当前元素，也就是最高的元素会被pop掉，所以凸出部分的面积，宽度会越来越宽

class Solution {
public:
    int largestRectangleArea(vector<int>& heights) {
        stack<int> st;
        heights.insert(heights.begin(), 0);
        heights.push_back(0);
        st.push(0);
        int result = 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 h = heights[mid];
                        int w = i - st.top() - 1;
                        result = max(result, h * w);
                    }
                }
                st.push(i);
            }
        }
        return result;
    }
};