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

Leetcode 739. 每日温度

题目链接
思路：寻找任一个元素的右边或者左边第一个比自己大或者小的元素的位置，此情景使用单调栈。
代码：

class Solution {
    public int[] dailyTemperatures(int[] temperatures) {
        /**
         * 如果当前遍历的元素大于栈顶元素，表示栈顶元素的右边的最大的元素就是当前遍历的元素，
         * 所以弹出栈顶元素，并记录，单调栈里存的是下标
         * 保持栈头到栈底单调递增
         */
        Stack<Integer> stack = new Stack<>();
        int[] result = new int[temperatures.length];
        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()]) {
                    result[stack.peek()] = i - stack.peek();
                    stack.pop();
                }
                stack.push(i);
            }
        }
        return result;
    }
}

Leetcode 496.下一个更大元素 I

题目链接
思路：单调栈
代码：

class Solution {
    public int[] nextGreaterElement(int[] nums1, int[] nums2) {
        Stack<Integer> stack = new Stack<>();
        int[] result = new int[nums1.length];
        Arrays.fill(result, -1);
        Map<Integer, Integer> map = new HashMap<>(nums1.length);
        // key:下标元素，value：下标
        for (int i = 0; i < nums1.length; i++) {
            map.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 (map.containsKey(nums2[stack.peek()])) {
                        Integer index = map.get(nums2[stack.peek()]);
                        result[index] = nums2[i];
                    }
                    stack.pop();
                }
                stack.add(i);
            }
        }
        return result;
    }
}