496. 下一个更大元素 I
难度简单
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 。
思路
- 下一个更大的元素，从左往右 => 从栈底到栈顶的单调递减栈
- 什么时候破坏单调减特性 => 遍历到的元素比栈顶元素大时
- 引入map，记录元素本身，和右侧第一个比它的元素
- 在
map中找nums1[i]对应的比它大的元素
最终代码
class Solution {
public int[] nextGreaterElement(int[] nums1, int[] nums2) {
// key: 元素本身 value: 右侧比它大的第一个元素
Map<Integer, Integer> map = new HashMap<>();
Deque<Integer> stack = new ArrayDeque<>();
for(int i = 0; i < nums2.length; i++) {
while(!stack.isEmpty() && nums2[stack.peek()] < nums2[i]) {
int index = stack.pop();
map.put(nums2[index], nums2[i]);
}
stack.push(i);
}
// 在map中找nums1[i]对应的值
int[] result = new int[nums1.length];
Arrays.fill(result, -1);
for(int i = 0; i < nums1.length; i++) {
int key = nums1[i];
if(map.containsKey(key)) {
result[i] = map.get(key);
}
}
return result;
}
}
