Leetcode刷题java之496.下一个更大的元素

待我称王封你为后i 2021-10-30 05:06 238阅读 0赞

执行结果：

通过

显示详情

执行用时 :9 ms, 在所有 Java 提交中击败了71.93%的用户

内存消耗 :38.8 MB, 在所有 Java 提交中击败了33.84%的用户

题目：

给定两个没有重复元素的数组 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。  
注意:

nums1和nums2中所有元素是唯一的。  
nums1和nums2 的数组大小都不超过1000。

思路：

利用栈将nums2中下一个更大的元素建立一个映射关系，然后 遍历nums1就可以得到结果了，那么利用栈是如何找到下一个更大元素的呢？我们模拟一下这个过程就可以看出来了。

比如开始进入一个4，入栈，然后再进入一个2，比4小，所以不是更的元素，还要入栈，在进入一个5,比栈顶的2大，所以2的下一个更大的元素是5，继续判断5比栈顶的4大，所以4的下一个更大的元素是5，栈为空，执行下面，5入栈，重复之前的操作，最后栈中会剩下一些没有下一个更大元素的，把他们置定为-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 x:nums2)
            {
                while(!stack.empty()&&stack.peek()<x)
                {
                    map.put(stack.pop(),x);
                }
                stack.push(x);
            }
            //栈中剩余的内容是没有下一个最大元素的，将他们的值置定为题目要求的-1
            while(!stack.empty())
            {
                map.put(stack.pop(),-1);
            }
            //根据映射关系计算结果
            for(int i=0;i<nums1.length;i++)
            {
                result[i]=map.get(nums1[i]);
            }
            return result;
        }
    }