【6.16 代随_59day】 下一个更大元素 II、接雨水

---

下一个更大元素 II

力扣连接：503. 下一个更大元素 II（中等）

单调栈的方法

加多一个for循环即可，和每日温度题几乎一样

代码

class Solution {
    public int[] nextGreaterElements(int[] nums) {
        Deque<Integer> stack = new LinkedList<>();
        int len = nums.length;
        int[] result = new int[len];
        Arrays.fill(result, -1);
        stack.addLast(0);

        for(int i=1;i<len;i++){
            while(!stack.isEmpty() && nums[i]>nums[stack.getLast()]){
                result[stack.getLast()] = nums[i]; 
                stack.removeLast();
            }

            stack.add(i);
        }

        for(int i=0;i<len;i++){
            while(!stack.isEmpty() && nums[i]>nums[stack.getLast()]){
                result[stack.getLast()] = nums[i]; 
                stack.removeLast();
            }
        }

        return result;

    }
}

---

---

接雨水

力扣连接：42. 接雨水（中等）

单调栈方法

以下逻辑主要就是三种情况

情况一：当前遍历的元素（柱子）高度小于栈顶元素的高度 height[i] < height[st.top()]
情况二：当前遍历的元素（柱子）高度等于栈顶元素的高度 height[i] == height[st.top()]
情况三：当前遍历的元素（柱子）高度大于栈顶元素的高度 height[i] > height[st.top()]

图解步骤

关键点：

• 遇到相同高度的柱子。更新栈内下标，就是将栈里元素（旧下标）弹出，将新元素（新下标）加入栈中。

代码

class Solution {
    public int trap(int[] height) {
        Deque<Integer> stack = new LinkedList<>();
        int len = height.length;
        int sum = 0;
        stack.addLast(0);

        for(int i=1;i<len;i++){
            if(height[i]<height[stack.getLast()]){
                stack.addLast(i);
            }else if(height[i]==height[stack.getLast()]){
                stack.removeLast();
                stack.addLast(i);
            }else{
                while(!stack.isEmpty() && height[i]>height[stack.getLast()]){
                    int mid = stack.getLast();
                    stack.removeLast();
                    if(!stack.isEmpty()){
                        int h = Math.min(height[i], height[stack.getLast()])-height[mid];
                        int w = i-stack.getLast()-1;
                        sum += h*w;
                    }
                }
                stack.addLast(i);
            }
        }

        return sum;
    }
}

动态规划方法

分别求当前水柱 左边和右边的最大值。

代码

class Solution {
    public int trap(int[] height) {
        int len = height.length;

        int[] left = new int[len];
        int[] right = new int[len];
        
        left[0] = 0;
        for(int i=1;i<len;i++){
            left[i] = Math.max(height[i-1], left[i-1]);
        }

        right[len-1] = 0;
        for(int i=len-2;i>=0;i--){
            right[i] = Math.max(height[i+1], right[i+1]);
        }
        int result = 0;
        for(int i=0;i<len;i++){
            int min = Math.min(left[i], right[i]);
            if(height[i]<min){
                result += min-height[i];
            }

        }

        return result;
    }
}

---

---