下一个更大元素

Posted 2022-12-03 vector6_

下一个更大元素

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