下一个更大元素 I 力扣 HashMap + Deque栈 JAVA

nums1 中数字 x 的 下一个更大元素 是指 x 在 nums2 中对应位置 右侧 的 第一个 比 x 大的元素。

给你两个 没有重复元素 的数组 nums1 和 nums2 ，下标从 0 开始计数，其中nums1 是 nums2 的子集。

对于每个 0 <= i < nums1.length ，找出满足 nums1[i] == nums2[j] 的下标 j ，并且在 nums2
确定 nums2[j] 的 下一个更大元素 。如果不存在下一个更大元素，那么本次查询的答案是 -1 。

返回一个长度为 nums1.length 的数组 ans 作为答案，满足 ans[i] 是如上所述的 下一个更大元素 。

示例 1：

输入：nums1 = [4,1,2], nums2 = [1,3,4,2].
输出：[-1,3,-1]
解释：nums1中每个值的下一个更大元素如下所述：

• 4 ，用加粗斜体标识，nums2 = [1,3,4,2]。不存在下一个更大元素，所以答案是 -1 。
• 1 ，用加粗斜体标识，nums2 = [1,3,4,2]。下一个更大元素是 3 。
• 2 ，用加粗斜体标识，nums2 = [1,3,4,2]。不存在下一个更大元素，所以答案是 -1 。

示例 2：

输入：nums1 = [2,4], nums2 = [1,2,3,4].
输出：[3,-1]
解释：nums1中每个值的下一个更大元素如下所述：

• 2 ，用加粗斜体标识，nums2 = [1,2,3,4]。下一个更大元素是 3 。
• 4 ，用加粗斜体标识，nums2 = [1,2,3,4]。不存在下一个更大元素，所以答案是 -1 。

提示：

1 <= nums1.length <= nums2.length <= 1000
0 <= nums1[i], nums2[i] <= 104
nums1和nums2中所有整数 互不相同
nums1 中的所有整数同样出现在 nums2 中

进阶：你可以设计一个时间复杂度为 O(nums1.length + nums2.length) 的解决方案吗？

解题思路：

1、利用空间换时间

2、对nums2进行处理，通过map存储nums2[i]的下一个最大元素nums2[j],如果没有就存储-1

3、需要用到栈，当栈顶元素存在一个更大的元素，那就将它取出，否者继续装入

4、由于有栈的辅助，栈内的元素从栈尾到栈顶时刻是由大到小排列的，换成队列是不行的，真是美妙

4、最后遍历完后栈内的元素说明它们没有更大元素，那就在map中对应赋值-1

代码:

class Solution {
    public int[] nextGreaterElement(int[] nums1, int[] nums2) {
           Map<Integer, Integer> map = new HashMap<>();//哈希表
           Deque stack = new ArrayDeque<Integer>();//栈
           
           for(int i = 0; i < nums2.length; i ++) {
        	   while(!stack.isEmpty() && nums2[i] > (int)stack.peek()) {
        		   map.put((Integer)stack.poll(), nums2[i]);
        	   }
        	   stack.push(nums2[i]);
           }
           
           while(!stack.isEmpty()) map.put((Integer)stack.poll(), -1);
           int len = nums1.length;
           int res[] = new int[len];
           for(int i = 0; i < len; i ++) res[i] = map.get(nums1[i]);
           return res; 
    }
}