算法刷题｜503.下一个更大元素Ⅱ、42.接雨水

下一个更大元素Ⅱ

题目：给定一个循环数组 nums （ nums[nums.length - 1] 的下一个元素是 nums[0] ），返回 nums 中每个元素的 下一个更大元素 。

数字 x 的 下一个更大的元素 是按数组遍历顺序，这个数字之后的第一个比它更大的数，这意味着你应该循环地搜索它的下一个更大的数。如果不存在，则输出 -1 。

思路：本题就在于怎么处理环的问题，可以使用取余数的方式来处理环

class Solution {
    public int[] nextGreaterElements(int[] nums) {
        int[] res = new int[nums.length];
        Arrays.fill(res,-1);
        if(nums.length == 0) return res;
        // 单调递增栈
        Deque<Integer> deque = new LinkedList<>();
        deque.addLast(0);
        // 因为是环，所以我们遍历的大小为数组长度的二倍，这样才能真正找到当前元素下一个更大的元素
        for(int i = 1;i<nums.length * 2;i++){
            // 因为是环，所以我们可以通过取余数来实现环
            int ii = i % nums.length;
            if(nums[ii] <= nums[deque.peekLast()]){
                deque.addLast(ii);
            }else{
                while(deque.size() > 0 && nums[ii] > nums[deque.peekLast()]){
                    // 收集结果
                    res[deque.peekLast()] = nums[ii];
                    // 将栈顶弹出
                    deque.pollLast();
                }
                // 当前元素小于了栈顶元素，就放进去
                deque.addLast(ii);
            }
        }
        return res;
    }
}

接雨水

题目：给定 n 个非负整数表示每个宽度为 1 的柱子的高度图，计算按此排列的柱子，下雨之后能接多少雨水。

先记录一下思路，后面在来具体看看代码
思路：我们维护一个单调递增的栈，然后遍历数组，如果遍历的当前元素大于栈顶元素，那么我们就可以计算面积了，高就是当前元素和栈顶下一个元素的最小值，宽就是当前元素和栈顶的下一个元素的距离

class Solution {
    public int trap(int[] height) {
        Deque<Integer> deque = new LinkedList<>();
        deque.addLast(0);
        int res = 0;
        for(int i = 1;i<height.length;i++){
            // 维护一个单调递增栈
            if(height[i] < height[deque.peekLast()]){
                deque.addLast(i);
            }else if(height[i] == height[deque.peekLast()]){
                // 如果遇到相同的，这里可以先弹出让后在入栈，这样就会少了一个没有必要的计算
                // 也可以不用弹出，直接入栈，只是下面计算的时候因为元素相同所有 h = 0，所有最后的面积也是0
                deque.pollLast();
                deque.addLast(i);
            }else{
                while(!deque.isEmpty() && height[i] > height[deque.peekLast()]){
                    // 当前遍历的元素就是右边柱子
                    // 栈顶就是中间柱子
                    // 弹出栈顶之后的栈顶就是左边柱子
                    
                    // 记录栈顶元素，因为要被弹出去了
                    int mid = deque.pollLast();
                    if(!deque.isEmpty()){
                        // 雨水的高度
                        int h = Math.min(height[i],height[deque.peekLast()]) - height[mid];
                        // 雨水的宽度
                        int w = i - deque.peekLast() - 1;
                        res += h * w;
                    }
                }
                deque.addLast(i);
            }
        }
        return res;
    }
}