代码随想录算法训练营第五十八天|739. 每日温度 、496.下一个更大元素 I

代码随想录算法训练营第五十八天|739. 每日温度 、496.下一个更大元素 I

每日温度

739. 每日温度
文章讲解：https://programmercarl.com/0739.%E6%AF%8F%E6%97%A5%E6%B8%A9%E5%BA%A6.html
题目链接：https://leetcode.cn/problems/daily-temperatures/
视频讲解：https://www.bilibili.com/video/BV1my4y1Z7jj/

自己看到题目的第一想法

暴力解法，双层循环。然后执行超时了。

public int[] dailyTemperatures(int[] temperatures) {
    int[] result = new int[temperatures.length];
    for(int i = 0; i < temperatures.length - 1; i++){
        int num = 0;
        boolean hasGT = false;
        for(int j = i + 1; j < temperatures.length; j++){
            num++;
            if(temperatures[j] > temperatures[i]){
                hasGT = true;
                break;
            }
        }
        if(hasGT){
            result[i] = num;
        }
    }
    return result;
}

看完代码随想录之后的想法

使用栈结构，往栈里push index而不是值。如果遇到比栈顶大的，就pop弹出，如果比栈顶小的，就压栈。
没看答案代码，自己按照思路写的：

public int[] dailyTemperatures(int[] temperatures) {
     //单调栈的处理方式，非单调就弹出
     // push的时候先peek一下看看是否小，小的话就push进去，大的话就pop完然后数一下数量，直接往结果集里加
     Stack<Integer> stack = new Stack<>();
     int[] result = new int[temperatures.length];
     // temperatures = [73,74,75,71,69,72,76,73]
     for(int i = 0; i < temperatures.length; i++){
         if(i == 0){
             stack.push(i); // stack = [73];
             continue;
         }
         while(!stack.isEmpty() && temperatures[i] > temperatures[stack.peek()]){
             int topIndex = stack.peek();//[0]
             //System.out.println("i:" + i +"temperatures:" + temperatures[i] + ",topIndex:" + topIndex + ",stack.peek:" + stack.peek());
             result[topIndex] = i - topIndex;
             stack.pop();
         }
         stack.push(i);
     }
     while(!stack.isEmpty()){
         result[stack.pop()] = 0;
     }
     return result;
 }

自己实现过程中遇到哪些困难

代码细节有点错误，temperatures[i] > temperatures[stack.peek()]写成了temperatures[i] > stack.peek()。
后面输出后排查到问题并修复。

下一个更大元素 I

496.下一个更大元素 I
文章讲解：https://programmercarl.com/0496.%E4%B8%8B%E4%B8%80%E4%B8%AA%E6%9B%B4%E5%A4%A7%E5%85%83%E7%B4%A0I.html
题目链接：https://leetcode.cn/problems/next-greater-element-i/
视频讲解：https://www.bilibili.com/video/BV1jA411m7dX/

自己看到题目的第一想法

按照上一题的单调栈思路去写。

看完代码随想录之后的想法

第一次读题目的时候没理解题意。该题目题意是nums1是nums2的子集，然后求解的是nums1的数字对应在nums2的数字位置往右的下一个大于当前的数字。如nums1=[1,2,4]；nums2=[2,3,1,4] 这里nums1的2对应的是nums2的第一个2的位置，然后求的是第一个2的位置的后一个大于2的数字。
整体思路：

1. 因为题目里给到nums1是nums2的子集，并且数组中的元素都不重复。因此可以定义一个map；key-值，value-index来记录nums1的内容。
2. 遍历nums2数组，用单调栈来处理栈内的元素，如果栈顶元素小于遍历到的当前的值，那可以将当前值作为nums1数字对应index的值。
   2.1. 如遍历nums2，第一个2先push到栈内，遇到第二个3时，发现3大于2，那就去找2对应的nums1里的位置，并将值添加到结果集中。

自己实现过程中遇到哪些困难

无

public int[] nextGreaterElement(int[] nums1, int[] nums2) {
    Map<Integer,Integer> nums1ByIndex = new HashMap<>();
    for(int i = 0; i < nums1.length; i++){
        nums1ByIndex.put(nums1[i],i);
    }
    int[] result = new int[nums1.length];
    Arrays.fill(result,-1);
    Stack<Integer> st = new Stack<>();
    st.push(nums2[0]);
    for(int i = 1;i < nums2.length; i++){
        // 如果栈不为空，且栈顶小于当前值。则可以找出栈定对应的value
        while(!st.isEmpty() && nums2[i] > st.peek()){
            if(nums1ByIndex.get(st.peek()) != null){
                result[nums1ByIndex.get(st.peek())] = nums2[i];
            }
            st.pop();
        }
        st.push(nums2[i]);
    }
    return result;
}

今日收获&学习时长

单调栈回顾了一下，单调栈的核心只考虑相邻两个值的比较，之前push过的就不管了，由后面的pop处理回调。（这里文字描述有点抽象，后面写题的时候才能回顾到）