解题记录 LeetCode 下一个更大元素 单调栈

题目链接: https://leetcode-cn.com/problems/next-greater-element-i/
题意: 给你两个 没有重复元素 的数组 nums1 和 nums2 ，其中nums1 是 nums2 的子集。
请你找出 nums1 中每个元素在 nums2 中的下一个比其大的值。
nums1 中数字 x 的下一个更大元素是指 x 在 nums2 中对应位置的右边的第一个比 x 大的元素。如果不存在，对应位置输出 -1 。
思路:
由于数组nums1是nums2的子集，整体思路是先求解出数组nums2中每个元素对应的下一个更大元素。然后存入map中，其中key是nums2中的元素值，value是下一个更大的元素。最后，遍历数组nums1，拿当前考察的元素作为key在map中获取value即可。
至此，关键问题是如何求解出数组nums2中每个元素对应的下一个更大元素。
我们逆序遍历数组nums2，对于当前考察的元素nums[i]来说：
如果栈不为空且nums[i]大于等于栈顶元素，说明栈中元素不会是其之前某个元素的下一个更大元素，所以将栈中元素出栈。原因在于，这里是逆序遍历，如果当前元素nums[i]大于栈中元素，那么对于其前面的元素来说，肯定nums[i]更可能是其下一个更大元素。
如果栈不为空且nums[i]小于栈顶元素，由于是逆序遍历，说明栈顶元素是nums[i]的下一个更大元素，所以将nums[i]作为key，栈顶元素作为value存入map。
如果栈为空，说明nums[i]没有下一个更大元素，所以将nums[i]作为key，-1作为value存入map。
最后，将nums[i]存入map，看其是否可能是之前某个元素的下一个更大元素。

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