Problem: 496. 下一个更大元素 I
思路
首先我们需要去用一个map保存在nums2中每一个元素及其对应的下一个更大的元素的值
例如:nums1 = [4,1,2], nums2 = [1,3,4,2]
在map中会保存:(1,3),(3,4)、
通过单调栈的方式实现这一需求,单调栈中存放的是数组nums2的下标值,每一次都去比较当前元素和上一个元素哪一个大,符合大就存到map里面,然后最后使用数组nums1中的值作为map的key一个一个去和map中的vale去找,能找到返回找到的,没有则是-1,因为nums1中的数字和nums中是肯定有一 一 映射的,除非一个都没有就全是-1。
解题方法
描述你的解题方法
复杂度
- 时间复杂度:
添加时间复杂度, 示例: O ( n ) O(n) O(n)
- 空间复杂度:
添加空间复杂度, 示例: O ( n ) O(n) O(n)
Code
class Solution {
public int[] nextGreaterElement(int[] nums1, int[] nums2) {
Deque<Integer> stack = new LinkedList();
Map<Integer, Integer> map = new HashMap();
for(int i = 0;i<nums2.length;i++){
while(!stack.isEmpty() && nums2[stack.getLast()]<nums2[i]){
int top = stack.pollLast();
map.put(nums2[top],nums2[i]);
}
stack.addLast(i);
}
int[] res = new int[nums1.length];
for(int i = 0;i<nums1.length;i++){
res[i] = map.getOrDefault(nums1[i],-1);
}
return res;
}
}