单调栈第二天

目录

503.下一个更大元素II

看到题目的第一想法

看到代码随想录的想法

自己实现时出现的困难

42. 接雨水

看到题目的第一想法

看到代码随想录的第一想法

自己实现时遇到的困难

---

503.下一个更大元素II

力扣题目链接(opens new window)

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

示例 1:

• 输入: [1,2,1]
• 输出: [2,-1,2]
• 解释: 第一个 1 的下一个更大的数是 2；数字 2 找不到下一个更大的数；第二个 1 的下一个最大的数需要循环搜索，结果也是 2。

看到题目的第一想法

        利用昨天的单调栈的思路来写

        但是它是循环数组该怎么解决？

        遍历完一次后再遍历一次，将栈中剩余的弹出

        最后剩余的赋值为-1

        成功做出来

看到代码随想录的想法

        没有我这么麻烦，for循环范围放大为2*length，nums[i%length]取模，逻辑和之前相同

        若当前比栈里的小则直接存入

        若当前比栈里的大，则弹出，栈顶下标对应的第一个最大值为栈里的值

自己实现时出现的困难

        把result全部赋值为-1 自己就不用处理-1了

class Solution {
    public int[] nextGreaterElements(int[] nums) {
        //也是用单调栈来写？
        //for循环两次到末尾
        Stack<Integer> stack = new Stack<>();
        //需要遍历到当前下标的前一个位置？
        stack.push(0);
        int[] result = new int[nums.length];
        for(int i=1;i<nums.length;i++){
            if(nums[stack.peek()]>=nums[i]){
                stack.push(i);
            }else{
                while(!stack.isEmpty()&&nums[stack.peek()]<nums[i]){
                    result[stack.peek()] = nums[i];
                    stack.pop();
                }
                //不要忘了把这个值也push进去
                stack.push(i);
            }
        }
        //此时stack中残留的元素:
        //1 nums[i]末尾的元素
        //2 比较大的几个元素
        if(!stack.isEmpty()){
        for(int i=0;i<nums.length;i++){            
            while(!stack.isEmpty()&&nums[stack.peek()]<nums[i]){
                    //弹出来的下标对应result 的下标
                    result[stack.peek()] = nums[i];
                    stack.pop();
            }
        }
        }
        while(!stack.isEmpty()){
            result[stack.peek()] = -1;
            stack.pop();
        }
        return result;
    }
}

42. 接雨水

力扣题目链接(opens new window)

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

示例 1：

• 输入：height = [0,1,0,2,1,0,1,3,2,1,2,1]
• 输出：6
• 解释：上面是由数组 [0,1,0,2,1,0,1,3,2,1,2,1] 表示的高度图，在这种情况下，可以接 6 个单位的雨水（蓝色部分表示雨水）。

看到题目的第一想法

        用单调栈，没想清楚处理逻辑，要怎么计算雨水

看到代码随想录的第一想法

       用单调栈，

       遇到第一个比栈顶大的元素时，弹出栈顶做记录，再利用当前遍历到的值计算差值和下标差值

        栈顶值(中间值)和栈顶的第二个值(前一个比栈顶大的)，当前下标(当前比栈顶大的)

        

        每次计算的是以栈顶值为底部能存放的雨水容量，累加起来

        选择两边的最小值，然后再与栈底的差值做计算

自己实现时遇到的困难

         if(height[i]==height[stack.peek()])

        遍历时，当遇到栈顶与当前数值相等时，进栈的应该是第二个元素，因为要考虑下标

class Solution {
    public int trap(int[] height) {
        //卡哥思路 ，单调栈
        //遇到第一个比当前大的，则弹出第一个，记录，再弹出第二个，取弹出的第二个和当前的最小值，同时下标的差值进行雨水面积计算
        Stack<Integer> stack = new Stack<>();
        int startIndex = 0;
        while(startIndex<height.length-1&&height[startIndex]<=height[startIndex+1]){
            startIndex++;
        }
        if(startIndex==height.length-1){
            return 0;
        }
        stack.push(startIndex);
        int maxArea = 0;
        for(int i=startIndex+1;i<height.length;i++){
            if(height[i]<height[stack.peek()]){
                stack.push(i);
            }else if(height[i]==height[stack.peek()]){
                //不能直接continue，或者说如果前面和后面相等，必须push后面的那个，比如说
                // [9,6,8,8,5,6,3] 这两个相等的8 必须push 后面这个8的下标
                //弹出前面那个
                stack.pop();
                stack.push(i);
                continue;
            }else{
                //弹出第一个栈顶,记录
                //用第二个栈顶来进行计算，取当前heit[i]和第二个栈顶的最小值，来进行计算
                while(!stack.isEmpty()&&height[stack.peek()]<height[i]){
                        //弹出第一个
                        int topOne = stack.pop();
                        //高度为 当前高度(第二低点，于当前选中元素的最小值)减去 最低点
                        if(!stack.isEmpty()){
                        int h = Math.min(height[i],height[stack.peek()])-height[topOne];
                        maxArea += h*(i-stack.peek()-1);
                        }   
                }
                stack.push(i);
            }
        }
        return maxArea;
    }
}