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;
    }
}