【LeetCode-496】496. 下一个更大元素 I

496. 下一个更大元素 I

题目描述

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

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

最优思路：单调栈+哈希

通过Stack、HashMap解决

• 先遍历大数组nums2，首先将第一个元素入栈；
• 继续遍历，当当前元素小于栈顶元素时，继续将它入栈；当当前元素大于栈顶元素时，栈顶元素出栈，此时应将该出栈的元素与当前元素形成key-value键值对，存入HashMap中；
• 当遍历完nums2后，得到nums2中元素所对应的下一个更大元素的hash表；
• 遍历nums1的元素在hashMap中去查找‘下一个更大元素’，当找不到时则为-1。

class Solution {
    public int[] nextGreaterElement(int[] nums1, int[] nums2) {
        Stack<Integer> stack = new Stack<>();
        HashMap<Integer, Integer> map = new HashMap<>();

        int []result = new int[nums1.length];

        for(int i = 0; i < nums2.length; i++) {
            while(!stack.empty() && nums2[i] > stack.peek()) {
                int topNum = stack.pop();
                // 找到了第一个大于当前栈顶元素的值，即找到了当前栈顶的下一个更大的值，将这种映射关系存放到map中
                map.put(topNum, nums2[i]);
            }
            stack.push(nums2[i]);
        }

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

        return result;
    }
}

同类型题目参考其他博客