代码随想录算法训练营-day59-503.下一个更大元素II、42. 接雨水

503. 下一个更大元素 II

1. 学习文章链接：
2. 思路：在每日温度的基础上，单一数组变成了循环数组。注意循环数组的处理手法。
3. 代码：

class Solution {
    public int[] nextGreaterElements(int[] nums) {
        if (nums == null || nums.length <= 1) return new int[]{-1};
        int len = nums.length;
        int[] res = new int[len];
        Arrays.fill(res, -1);
        Stack<Integer> st = new Stack<>();
        for (int i = 0; i < len * 2; i++) {
            while (!st.isEmpty() && nums[i % len] > nums[st.peek()]) {
                res[st.peek()] = nums[i % len];
                st.pop();
            }
            st.push(i % len);
        }
        return res;
    }
}

42. 接雨水

1. 学习文章链接：
2. 思路：优先理解单调栈的解法。
3. 代码：

class Solution {
    public int trap(int[] height) {
        int len = height.length;
        if (len <= 2) return 0;
        Stack<Integer> st = new Stack<>();
        st.push(0);
        int sum = 0;
        for (int i = 1; i < len; i++) {
            if (height[i] < height[st.peek()]) {
                st.push(i);
            } else if (height[i] == height[st.peek()]) {
                st.pop();
                st.push(i);
            } else {
                while (!st.isEmpty() && height[i] > height[st.peek()]) {
                    int mid = st.pop();
                    if (!st.isEmpty()) {
                        int left = st.peek();

                        int h = Math.min(height[left], height[i]) - height[mid];
                        int w = i - left - 1;
                        if (h * w > 0) sum += h * w;
                    }
                }
                st.push(i);
            }
        }
        return sum;
    }
}