leetcode-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 。
思路
阅读题目后,得知要找出与num2数组中对应num1数组的值,并且要求得这些值在num2数组中的下一个比这些值更大的数,找数组中比目标值更大的数,这显然是单调栈的拿手活儿,所以基本思路可以确定使用单调栈解决,但这题跟739.每日温度那题不同之处在于,它外层还需要找到num2与num1数组中相等的值,才能确定要找哪些目标值,显然按这个思路外层还得加一层循环,时间复杂度就来到了O(m*n)了,但是观察题目有这么一句:
进阶:你可以设计一个时间复杂度为
O(nums1.length + nums2.length)
的解决方案吗?
所以要保证达到这个复杂度肯定不能再加for,这时候可以使用Map来记录num2中获得的值以及目标值,即:Map中的K存储当前元素的值,V为计算得出的下一个比它大的值,这样就能在最外围一层循环中通过num1[K]直接获取Value并逐个赋值给res结果数组
代码
class Solution {
public static int[] nextGreaterElement(int[] nums1, int[] nums2) {
Stack<Integer> stack = new Stack<>();
Map<Integer, Integer> map = new HashMap<>(); // K为当前元素的值,V为计算得出的下一个比它大的值
for (int i = 0; i < nums2.length; i++) {
while(!stack.isEmpty() && nums2[i] > nums2[stack.peek()]){
Integer currPop = stack.pop();
map.put(nums2[currPop],nums2[i]);
}
stack.push(i);
}
while(!stack.isEmpty()){
map.put(nums2[stack.pop()],-1);
}
int[] res = new int[nums1.length];
for (int i = 0; i < nums1.length; i++) {
res[i] = map.get(nums1[i]);
}
return res;
}
}
本文是原创文章,采用 CC BY-NC-ND 4.0 协议,完整转载请注明来自 DoublePeach
评论
匿名评论
隐私政策
你无需删除空行,直接评论以获取最佳展示效果