下一个更大元素-CSDN博客

本文链接：https://blog.csdn.net/qq_41105501/article/details/124512307

下一个更大元素

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] 是如上所述的下一个更大元素。

使用单调栈逆序遍历num2，得到nums2中当前位置处下一个更大的元素值。使用Map保存该元素对应的目标值

class Solution {
public:
    vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {
        vector<int> res;
        std::stack<int> st;
        std::unordered_map<int, int> numMap;
        for(int i = nums2.size()-1; i>=0; --i)
        {
            while(!st.empty() && nums2[i]>=st.top())
            {
                st.pop();
            }
            if(!st.empty())
            {
                numMap.insert(make_pair(nums2[i],st.top()));
            }
            else
            {
                numMap.insert(make_pair(nums2[i],-1));
            }
            st.push(nums2[i]);
        }
        for(int i=0;i<nums1.size(); ++i)
        {
            res.push_back(numMap[nums1[i]]);
        }
        return res;
    }
};