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

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

什么情况下使用单调栈：通常是一维数组，要寻找任一个元素的右边或者左边第一个比自己大或者小的元素的位置，此时我们就要想到可以用单调栈了。时间复杂度为O(n)。

一、力扣739. 每日温度

题目链接
思路：使用单调栈存放遍历过的元素，栈内自栈顶到栈底单调递增，只要当前元素比栈顶元素大，就找到了栈顶元素右边第一个比它大的值，当前元素比栈顶元素小，直接入栈即可，等到后面又遍历到更大的数，就算是找到了之前小的的右边第一个大于他的值。

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;
    }
}

二、力扣496.下一个更大元素 I

题目链接
思路：这个要找的是比它更大的下一个值，不是索引，两个数组顺序不同使用HashMap进行映射，只有当元素存在时才记录结果。

class Solution {
    public int[] nextGreaterElement(int[] nums1, int[] nums2) {
        int[] res = new int[nums1.length];
        Deque<Integer> stack = new LinkedList<>();
        Map<Integer, Integer> map = new HashMap<>();
        for (int i = 0; i < nums1.length; i++) {
            res[i] = -1;
            map.put(nums1[i], i);
        }
        stack.push(0);
        for (int i = 1; i < nums2.length; i++) {
            if (nums2[i] <= nums2[stack.peek()]) {
                stack.push(i);
            }else {
                while (!stack.isEmpty() && nums2[i] > nums2[stack.peek()]) {
                    if (map.containsKey(nums2[stack.peek()])) {
                        res[map.get(nums2[stack.peek()])] = nums2[i];
                    }
                    stack.pop();
                }
                stack.push(i);
            }
        }
        return res;
    }
}