单调栈
问题:下一个更大元素
给定两个没有重复元素的数组nums1和nums2,其中nums1是nums2的子集。找到nums1中每个元素在nums2中的下一个比其大的值。
nums1 中数字 x 的下一个更大元素是指 x 在 nums2 中对应位置的右边的第一个比 x 大的元素。如果不存在,对应位置输出-1。
例如:输入: nums1 = [4,1,2], nums2 = [1,3,4,2].
输出: [-1,3,-1]
单调栈解法
维护一个单调递增的栈。
从右向左依此遍历nums2数组,当遍历的数x比栈顶元素小时,入栈(此时找到了第一个比x大的数,即为栈顶元素),并在map中记录。当遍历的数x比栈顶元素大时,将栈顶元素出栈,然后将x入栈,在x入栈之前需要判断栈是否为空,如果栈不为空,则x的第一个最大元素就是栈顶元素,如果栈为空,则x的下一个最大元素就是-1.
然后将nums1中的值作为key依此查找map,得到结果。
代码实现:
public int[] nextGreaterElement(int[] nums1, int[] nums2) {
int[] res = new int[nums1.length];
Map<Integer,Integer> map = new HashMap<>();
Stack<Integer> s = new Stack<>();
for(int i=nums2.length-1;i>=0;i--){
while(!s.isEmpty() && nums2[i] > s.peek()){
s.pop();
}
if(s.isEmpty())
map.put(nums2[i],-1);
else{
map.put(nums2[i],s.peek());
}
s.push(nums2[i]);
}
for(int i=0;i<nums1.length;i++){
res[i] = map.get(nums1[i]);
}
return res;
}