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

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

739. 每日温度

class Solution {
    public int[] dailyTemperatures(int[] temperatures) {
        // 单调递增是这么写
        int len = temperatures.length;
        int[] result = new int[len];

        Deque<Integer> queue = new LinkedList<>();
        // 这个放的是下标
        queue.push(0);
        for (int i = 1; i < len; i++) {
            if (temperatures[i] <= temperatures[queue.peek()]) {
                queue.push(i);
            } else {
                // 遇到
                while (!queue.isEmpty() && temperatures[i]>temperatures[queue.peek()]){
                    // 这个才是重点
                    result[queue.peek()] = i - queue.peek();
                    queue.pop();
                }
                // 这个也是
                queue.push(i);
            }
        }

        return result;
    }
}

496.下一个更大元素 I

class Solution {
    public int[] nextGreaterElement(int[] nums1, int[] nums2) {
        // 单调栈 递增是4,3,2,1 递减是1,2,3,4
        //  我的第一想法就是模拟。题解是先去处理nums2
        int len1 = nums1.length;
        int len2 = nums2.length;
        int[] result = new int[len1];

        if (len1 == 0) {
            return result;
        }
        HashMap<Integer, Integer> map = new HashMap<>();
        for (int i = 0; i < len1; i++) {
            result[i] = -1;
            map.put(nums1[i],i);
        }

        Stack<Integer> stack = new Stack<>();
        // 放的是下标
        stack.push(0);
        for (int i = 1; i < len2; i++) {
            if (nums2[i] <= nums2[stack.peek()]) {
                stack.push(i);
            } else {
                while (!stack.isEmpty() && nums2[i] > nums2[stack.peek()]) {
                    // 说明在nums1中有这个元素
                    if (map.containsKey(nums2[stack.peek()])){
                        result[map.get(nums2[stack.peek()])] = nums2[i];
                    }
                    // 不管什么都要弹出，不然就不是单调栈了
                    stack.pop();
                }
                //不要忘记放入
                stack.push(i);
            }
        }
        return result;

    }
}