leetcode 496. 下一个更大元素 I
输入: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 。
解法:单调栈求一个数组中的下一个最大元素,用map保存对应元素。
class Solution {
public int[] nextGreaterElement(int[] nums1, int[] nums2) {
int len1=nums1.length,len2=nums2.length;
Map<Integer,Integer> map=new HashMap<>();
Deque<Integer> stack=new ArrayDeque<>();
for(int i=0;i<len2;i++){
while(!stack.isEmpty()&&nums2[i]>nums2[stack.peek()]){
map.put(nums2[stack.peek()],nums2[i]);
stack.pop();
}
stack.push(i);
}
int[] res=new int[len1];
for(int i=0;i<len1;i++){
res[i]=map.getOrDefault(nums1[i],-1);
}
return res;
}
}

浙公网安备 33010602011771号