(Leetcode 刷题) 下一个更大元素Ⅰ

题目描述

给定两个 没有重复元素 的数组 nums1 和 nums2 ，其中nums1 是 nums2 的子集。找到 nums1 中每个元素在 nums2 中的下一个比其大的值。

nums1 中数字 x 的下一个更大元素是指 x 在 nums2 中对应位置的右边的第一个比 x 大的元素。如果不存在，对应位置输出 -1 。

496. 下一个更大元素Ⅰ

解法

构造一个单调栈：从栈顶到栈底的元素单调不减，每当遇到一个比栈顶元素大的，说明栈中所有元素的下一个更大元素是当前元素，将所有元素出栈并存在HashMap里。

class Solution {
    public int[] nextGreaterElement(int[] nums1, int[] nums2) {
        HashMap map = new HashMap<>();
        LinkedList stack = new LinkedList<>();
        for(int a : nums2){
            //注意循环条件
            //栈为空没有比较必要，元素入栈；
            //栈不空但元素 a 比栈顶元素小，继续入栈元素；
            //栈不空且元素 a 比栈顶大，所有元素出栈，结果存入HashMap中
            while(!stack.isEmpty() && stack.peek() < a){
                map.put(stack.pop(), a);
            }
            stack.push(a);
        }

        int[] res = new int[nums1.length];
        for(int i = 0; i < res.length; i++){
            res[i] = map.getOrDefault(nums1[i], -1);
        }
        return res;
    }
}