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

题目来源：https://leetcode-cn.com/problems/next-greater-element-i/submissions/

大致题意：
给定两个数组，保证所有元素不重复，且前一个数组是后一个的子集
对于前一个数组的每个元素，若其在第二个数组的位置之后，有大于该元素的元素，那么返回第一个大于它的值，否则返回 -1

思路

• 暴力
• 递增栈 + 哈希表

递增栈 + 哈希表

• 倒序遍历，使用递增栈存下在当前位置之后比当前数大的所有数
• 用哈希表存当前数之后第一个更大的数

有了这两个数据结构，逆序遍历一次后一个数组，就可以得出所有元素的对应结果。然后遍历前一个数组，从哈希表中取出对应结果即可

这里需要主要的是更新递增栈的方法：

1. 对于当前元素，若栈不为空且栈顶元素小于当前元素，那么显然当前递增栈不满足要求，需要将栈顶元素抛出，重复这一步直至栈为空或者栈顶元素大于当前元素
2. 如果栈为空，那么显然在当前数之后没有更大的数，对应的结果为 -1；否则，对应的结果即为栈顶元素
3. 将当前元素入栈

代码：

public int[] nextGreaterElement(int[] nums1, int[] nums2) {
        Map<Integer, Integer> next = new HashMap<>();   // 存当前数之后第一个更大的数
        Deque<Integer> stack = new ArrayDeque<>();  // 递增栈，存在当前位置之后比当前数大的所有数
        for (int i = nums2.length - 1; i >= 0; i--) {
            // 若栈不为空，且栈顶元素小于当前元素，循环直至栈为空或栈顶元素更大
            while (!stack.isEmpty() && nums2[i] > stack.peek()) {   
                stack.pop();
            }
            // 得出结果
            int greater = stack.isEmpty() ? -1 : stack.peek();
            // 当前元素入栈
            stack.push(nums2[i]);
            // 当前计算结果放入哈希表
            next.put(nums2[i], greater);
        }
        int[] ans = new int[nums1.length];
        // 从哈希表取出对应的结果
        for (int i = 0; i < nums1.length; i++) {
            ans[i] = next.get(nums1[i]);
        }
        return ans;
    }