代码随想录第59天|503.下一个更大元素II ， 42. 接雨水

LeetCode503.下一个更大元素II 

题目链接：503. 下一个更大元素 II - 力扣（LeetCode）

思路：

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

LeetCode42. 接雨水

题目链接：42. 接雨水 - 力扣（LeetCode）

思路：

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()]) {// 情况1当前遍历的元素（柱子）高度小于栈顶元素的高度 height[i] < height[st.top()]
                st.push(i);
            }
            else if(height[i] == height[st.top()]) {//情况二：当前遍历的元素（柱子）高度等于栈顶元素的高度 height[i] == height[st.top()]
                st.pop();
                st.push(i);
            }
            else {//情况三：当前遍历的元素（柱子）高度大于栈顶元素的高度 height[i] > height[st.top()]
                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;
    }
};