LeetCode——496. 下一个更大元素 I(Java)

题目描述

题干:
给你两个 没有重复元素 的数组 nums1 和 nums2 ,其中nums1 是 nums2 的子集。
请你找出 nums1 中每个元素在 nums2 中的下一个比其大的值。
nums1 中数字 x 的下一个更大元素是指 x 在 nums2 中对应位置的右边的第一个比 x 大的元素。
如果不存在,对应位置输出 -1 。

示例 1:
输入: nums1 = [4,1,2], nums2 = [1,3,4,2].
输出: [-1,3,-1]
解释:
    对于 num1 中的数字 4 ,你无法在第二个数组中找到下一个更大的数字,因此输出 -1 。
    对于 num1 中的数字 1 ,第二个数组中数字1右边的下一个较大数字是 3 。
    对于 num1 中的数字 2 ,第二个数组中没有下一个更大的数字,因此输出 -1 。

示例 2:
输入: nums1 = [2,4], nums2 = [1,2,3,4].
输出: [3,-1]
解释:
    对于 num1 中的数字 2 ,第二个数组中的下一个较大数字是 3 。
    对于 num1 中的数字 4 ,第二个数组中没有下一个更大的数字,因此输出 -1 。

题目描述

返回数组中右边第一个最大的元素,各种结论里都是说遇到这种问题直接就是单调栈

那这里我们就不介绍暴力的解法了,我们来说说单调栈这个结构

单调栈我们分为单调递增栈和单调递减栈,他的结构见名知意就是里面的元素有规律的排序

无论是递增还是递减,都是我们自己人为决定的,这里我们要拿到右边较大的第一个元素

那我们就用递减栈,这样可以保证拿到的是比他大的最小元素,具体代码如下

正确代码

    // 单调栈和哈希表
    public int[] nextGreaterElement(int[] nums1, int[] nums2) {
        // 定义用到的栈
        Map<Integer, Integer> hashMap = new HashMap<>();
        Deque<Integer> arrayDeque = new ArrayDeque<>();

        // 反向遍历借助栈保存右侧最大值
        for (int i = nums2.length - 1; i >= 0; i--) {
            int num = nums2[i];

            while (!arrayDeque.isEmpty() && num >= arrayDeque.peek()) {
                arrayDeque.pop();
            }
            hashMap.put(num, arrayDeque.isEmpty() ? -1 : arrayDeque.peek());
            arrayDeque.push(num);
        }

        // 依次获得哈希中的值
        int[] ansArr = new int[nums1.length];
        for (int i = 0; i < nums1.length; i++) {
            ansArr[i] = hashMap.get(nums1[i]);
        }

        return ansArr;
    }

总结

对于栈结构来说,Java的实现方式有很多,但是最推荐的就是采用Deque来实现

之所以不采用Vector的Stack是因为他保证线程安全并且只能在尾部操作,效率太低

还有就是采用LinkList来实现,它和ArrayDeque就是见仁见智了

如果文章存在问题或者有更好的题解,欢迎在评论区斧正和评论,各自努力,最高处见
posted @ 2021-10-26 09:34  21岁还不是架构师  阅读(62)  评论(0编辑  收藏  举报