代码随想录算法训练营第五十九天|503.下一个更大元素II、42. 接雨水

503.下一个更大元素II

文档讲解 ： 代码随想录 - 503.下一个更大元素II
状态：再次回顾。

本题的难点在于：如何处理循环数组
可以尝试将两个数组拼接在一起

本题代码：

class Solution {
public:
    vector<int> nextGreaterElements(vector<int>& nums) {
        vector<int> result(nums.size(), -1);
        if (nums.size() == 0) return result;
        stack<int> st;
        for (int i = 0; i < nums.size() * 2; i++) {
            // 模拟遍历两边nums，注意一下都是用i % nums.size()来操作
            while (!st.empty() && nums[i % nums.size()] > nums[st.top()]) {
                result[st.top()] = nums[i % nums.size()];
                st.pop();
            }
            st.push(i % nums.size());
        }
        return result;
    }
};

42. 接雨水 ★

文档讲解 ： 代码随想录 - 42. 接雨水
状态：再次回顾。（★：需要多次回顾并重点回顾）

单调栈思路：

1. 首先单调栈是按照行方向来计算雨水，如图：
   
2. 单调栈内元素的顺序
   从栈头（元素从栈头弹出）到栈底的顺序应该是从小到大的顺序。
   • 因为一旦发现添加的柱子高度大于栈头元素了，此时就出现凹槽了，栈头元素就是凹槽底部的柱子，栈头第二个元素就是凹槽左边的柱子，而添加的元素就是凹槽右边的柱子。
     
3. 遇到相同高度的柱子如何处理
   遇到相同的元素，更新栈内下标，就是将栈里元素（旧下标） 弹出，将新元素（新下标） 加入栈中。
   因为我们要求宽度的时候 如果遇到相同高度的柱子，需要使用最右边的柱子来计算宽度。
   
4. 栈里要保存什么数值
   保存下标就行了，想要知道对应的高度，通过height[stack.top()] 就知道弹出的下标对应的高度了。

单调栈处理逻辑：
栈顶和栈顶的下一个元素以及要入栈的元素，三个元素来接水

• 雨水高度是 min(凹槽左边高度, 凹槽右边高度) - 凹槽底部高度，代码为：int h = min(height[st.top()], height[i]) - height[mid];
• 雨水的宽度是 凹槽右边的下标 - 凹槽左边的下标 - 1（因为只求中间宽度），代码为：int w = i - st.top() - 1 ;
• 当前凹槽雨水的体积就是：h * w。

本题代码：

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