【单调栈】|代码随想录算法训练营第41天|739. 每日温度、 496.下一个更大元素 I、503.下一个更大元素II

刷题神器

代码随想录

往期回顾

>【回溯算法】|代码随想录算法训练营第22天|491.递增子序列、 46.全排列、47.全排列 II、【总结】

题目

739. 每日温度

题目：题目链接
文章：文章讲解
视频：视频讲解

• 学后思路

  使用单调栈解决，注意单调栈里面的存放内容为下标值

解法1:

class Solution {
    public int[] dailyTemperatures(int[] temperatures) {
        int len = temperatures.length;
        int[] res = new int[len];

        Deque<Integer> stack = new LinkedList<>();
        stack.push(0);
        for (int i = 1; i < len; i++) {
            if (temperatures[i] <= temperatures[stack.peek()]) {
                stack.push(i);
            } else {
        /**
        如果当前遍历的元素 大于栈顶元素，表示 栈顶元素的 右边的最大的元素就是 当前遍历的元素，
        	所以弹出 栈顶元素，并记录
        	如果栈不空的话，还要考虑新的栈顶与当前元素的大小关系
        否则的话，可以直接入栈。
        注意，单调栈里 加入的元素是 下标。
        */
                while (!stack.isEmpty() && temperatures[i] > temperatures[stack.peek()]) {
                    res[stack.peek()] = i - stack.peek();
                    stack.pop();
                }
                stack.push(i);
            }
        }
        return res;

    }
}

• 题目总结
  • 注意单调递增还是递减
  • 注意栈的逻辑判断条件

496.下一个更大元素 I

题目：题目链接
文章：文章讲解
视频：视频讲解

• 学后思路
  解题思路还是使用单调栈，但是要注意两个数组如何映射

解法一:

class Solution {
    public int[] nextGreaterElement(int[] nums1, int[] nums2) {
        Stack<Integer> stack = new Stack<>();
        // 结果集是nums1的长度
        int[] res = new int[nums1.length];
        Arrays.fill(res, -1);
        // 保存nums1的 value index的映射过线
        HashMap<Integer, Integer> hashMap = new HashMap<>();
        for (int i = 0; i < nums1.length; i++) {
            hashMap.put(nums1[i], i);
        }
        stack.add(0);
        for (int i = 1; i < nums2.length; i++) {
            if (nums2[i] <= nums2[stack.peek()]) {
                stack.add(i);
            } else {
                while (!stack.isEmpty() && nums2[stack.peek()] < nums2[i]) {
                    if (hashMap.containsKey(nums2[stack.peek()])) {
                        // 判断 nums2中的值是否是nums1中的值 转换index
                        Integer index = hashMap.get(nums2[stack.peek()]);
                        res[index] = nums2[i];
                    }
                    stack.pop();
                }
                stack.add(i);
            }
        }

        return res;
    }
}

• 题目总结
  • 注意result的长度
  • 注意nums1与nums2的index转换
  • hashMap判断nums2中的值是否在nums1中

503.下一个更大元素II

题目：题目链接
文章：文章讲解
视频：视频讲解

• 学后思路
  循环数组，使用取模的方式进行循环

解法一:

class Solution {
    public int[] nextGreaterElements(int[] nums) {
        //边界判断
        if(nums == null || nums.length <= 1) {
            return new int[]{-1};
        }
        int size = nums.length;
        int[] result = new int[size];//存放结果
        Arrays.fill(result,-1);//默认全部初始化为-1
        Stack<Integer> st= new Stack<>();//栈中存放的是nums中的元素下标
        for(int i = 0; i < 2*size; i++) {
            while(!st.empty() && nums[i % size] > nums[st.peek()]) {
                result[st.peek()] = nums[i % size];//更新result
                st.pop();//弹出栈顶
            }
            st.push(i % size);
        }
        return result;
    }
}

• 题目总结
  • 注意循环是2倍的数组长度
  • 循环过程中不会发生覆盖的情况，比如234 或者 432，234产生的结果是一样的，432，后面栈中的结果没有处理，所以没有影响