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 。

提示：

1 <= nums1.length <= nums2.length <= 1000

0 <= nums1 $i$ , nums2 $i$ <= 104^44

nums1和nums2中所有整数互不相同

nums1 中的所有整数同样出现在 nums2 中

进阶：你可以设计一个时间复杂度为 O(nums1.length + nums2.length) 的解决方案吗？

可以使用单调栈，，每遍历到nums2中的一个元素，看它下一个更大元素是多少，如果它同时存在于nums1中，就将它的下一个更大元素加入答案：

cpp 复制代码

class Solution {
public:
    vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {
        unordered_map<int, int> vTok;
        for (int i = 0; i < nums1.size(); ++i) {
            vTok[nums1[i]] = i;
        }

        vector<int> ans(nums1.size(), -1);
        stack<int> stk;
        for (int i = nums2.size() - 1; i >= 0; --i) {
            // 找到比当前值大的下一个更大元素，循环结束后，该值要么不存在，要么在栈顶
            while (!stk.empty() && nums2[i] > stk.top()) {
                stk.pop();
            }

            // 如果存在下一个更大元素，且当前值也存在于nums1中
            if (!stk.empty() && vTok.find(nums2[i]) != vTok.end()) {
                ans[vTok[nums2[i]]] = stk.top();
            }

            stk.push(nums2[i]);
        }

        return ans;
    }
};

如果nums1的长度为n，nums2的长度为m，则此算法时间复杂度为O(n+m)，空间复杂度为O(n+m)。