题目链接: https://leetcode-cn.com/problems/next-greater-element-i/
题意: 给你两个 没有重复元素 的数组 nums1 和 nums2 ,其中nums1 是 nums2 的子集。
请你找出 nums1 中每个元素在 nums2 中的下一个比其大的值。
nums1 中数字 x 的下一个更大元素是指 x 在 nums2 中对应位置的右边的第一个比 x 大的元素。如果不存在,对应位置输出 -1 。
思路:
由于数组nums1是nums2的子集,整体思路是先求解出数组nums2中每个元素对应的下一个更大元素。然后存入map中,其中key是nums2中的元素值,value是下一个更大的元素。最后,遍历数组nums1,拿当前考察的元素作为key在map中获取value即可。
至此,关键问题是如何求解出数组nums2中每个元素对应的下一个更大元素。
我们逆序遍历数组nums2,对于当前考察的元素nums[i]来说:
如果栈不为空且nums[i]大于等于栈顶元素,说明栈中元素不会是其之前某个元素的下一个更大元素,所以将栈中元素出栈。原因在于,这里是逆序遍历,如果当前元素nums[i]大于栈中元素,那么对于其前面的元素来说,肯定nums[i]更可能是其下一个更大元素。
如果栈不为空且nums[i]小于栈顶元素,由于是逆序遍历,说明栈顶元素是nums[i]的下一个更大元素,所以将nums[i]作为key,栈顶元素作为value存入map。
如果栈为空,说明nums[i]没有下一个更大元素,所以将nums[i]作为key,-1作为value存入map。
最后,将nums[i]存入map,看其是否可能是之前某个元素的下一个更大元素。
class Solution {
public int[] nextGreaterElement(int[] nums1, int[] nums2) {
Map<Integer, Integer> map = new HashMap<>();
Deque<Integer> deque = new ArrayDeque<>();
int ans[] = new int[nums1.length];
for(int i = nums2.length - 1; i >= 0; i--) {
while(!deque.isEmpty() && deque.peek() < nums2[i]) {
deque.pop();
}
if(deque.isEmpty()) {
map.put(nums2[i], -1);
} else {
map.put(nums2[i], deque.peek());
}
deque.push(nums2[i]);
}
for(int i = 0; i < nums1.length; i++) {
ans[i] = map.get(nums1[i]);
}
return ans;
}
}