【代码随想录训练营】【Day58】第十章｜单调栈｜739.每日温度｜496.下一个更大元素 I

每日温度

题目详细：LeetCode.739

详细的题解可查阅：《代码随想录》— 每日温度

Java解法（单调栈）：

class Solution {
    public int[] dailyTemperatures(int[] temperatures) {
        Stack<Integer> stack = new Stack<>();
        int[] res = new int[temperatures.length];
        for(int i = 0; i < temperatures.length; i++){
            while(!stack.isEmpty() && temperatures[i] > temperatures[stack.peek()]){
                res[stack.peek()] = i - stack.pop();
            }
            stack.push(i);
        }
        return res;
    }
}

---

下一个更大元素 I

题目详细：LeetCode.496

详细的题解可查阅：《代码随想录》— 下一个更大元素 I

Java解法（单调栈）：

class Solution {
    public int[] nextGreaterElement(int[] nums1, int[] nums2) {
        // 定义一个哈希表<nums1中的数字，数字的下标>
        HashMap<Integer, Integer> map = new HashMap<>();
        for (int i = 0; i < nums1.length; i++) {
            map.put(nums1[i], i);
        }
        int[] res = new int[nums1.length];
        Arrays.fill(res, -1);
        Stack<Integer> stack = new Stack<>();
        // 从左往右遍历nums2，保证栈单调递减
        for (int i = 0; i < nums2.length; i++) {
            while (!stack.isEmpty() && nums2[stack.peek()] < nums2[i]) {
                // 当出现nums2[stack.peek()] < nums2[i]时，说明当前元素nums2[i]是栈内元素的下一个更大元素
                // 将栈内元素出栈
                int pre = nums2[stack.pop()];
                // 判断站内元素是否在nums1中存在，存在则更新其对应下标的值为下一个更大元素nums2[i]
                if (map.containsKey(pre)) {
                    res[map.get(pre)] = nums2[i];
                }
            }
            stack.push(i);
        }
        return res;
    }
}

---