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

leetcode 739 每日温度

题目链接

739. 每日温度 - 力扣（LeetCode）

做题过程

单调栈的思路是构建一个栈。‘

然后考虑三种情况：

• 当前遍历的元素T[i]小于栈顶元素T[st.top()]的情况
• 当前遍历的元素T[i]等于栈顶元素T[st.top()]的情况
• 当前遍历的元素T[i]大于栈顶元素T[st.top()]的情况

前两种直接入栈即可。问题在于第三种。

第三种情况我们需要比较temperatures[i] > temperatures[stack.peek()]。

若大于则说明stack.peek()位置的元素在第i天会递增。

因此计算两者差值即可，同时出栈。继续比较栈顶与i元素的值。

解决方法

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

        Deque<Integer> stack = new LinkedList<>();
        stack.push(0);

        for(int i = 1;i < temperatures.length; 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;

    }
}

leetcode 496 下一个更大元素 I

题目链接

496. 下一个更大元素 I - 力扣（LeetCode）

做题过程

与上一道题目思路几乎相同。但这次要反复比较nums1与nums2的值。因此用HashMap来记录方便快速查询。

解决方法

class Solution {
    public int[] nextGreaterElement(int[] nums1, int[] nums2) {
        Stack<Integer> temp = new Stack<>();
        int[] res = new int[nums1.length];
        Arrays.fill(res,-1);
        HashMap<Integer, Integer> hashMap = new HashMap<>();
        for (int i = 0 ; i< nums1.length ; i++){
            hashMap.put(nums1[i],i);
        }
        temp.add(0);
        for (int i = 1; i < nums2.length; i++) {
            if (nums2[i] <= nums2[temp.peek()]) {
                temp.add(i);
            } else {
                while (!temp.isEmpty() && nums2[temp.peek()] < nums2[i]) {
                    if (hashMap.containsKey(nums2[temp.peek()])){
                        Integer index = hashMap.get(nums2[temp.peek()]);
                        res[index] = nums2[i];
                    }
                    temp.pop();
                }
                temp.add(i);
            }
        }

        return res;
    }
}

​