单调栈系列-Leetcode 496. 下一个更大元素 I

问题描述

给定两个 没有重复元素 的数组 nums1 和 nums2 ，其中nums1 是 nums2 的子集。找到 nums1 中每个元素在 nums2 中的下一个比其大的值。

nums1 中数字 x 的下一个更大元素是指 x 在 nums2 中对应位置的右边的第一个比 x 大的元素。如果不存在，对应位置输出 -1 $^{[1]}$。

解题报告

求解每个元素的下一个更大元素，采用 单调递减栈 来存储数组中的元素，当前元素小于栈顶元素时，压栈；当前元素大于栈顶元素时，当前元素即为栈顶元素的下一个更大元素，出栈，记录栈顶元素的下一个更大元素，直到栈顶元素小于当前元素。

将每个元素的下一个更大元素存到哈希表中，之后可以在 $O(1)$ 的时间复杂度内查询。

实现代码

class Solution {
public:
    vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {
        unordered_map<int, int>mp;
        stack<int>stk;
        for(int i=0;i<nums2.size();i++){
            while(!stk.empty()&&nums2[stk.top()]<nums2[i]){
                mp[nums2[stk.top()]]=nums2[i];
                stk.pop();
            }
            stk.push(i);
        }

        for(int i=0;i<nums1.size();i++){
            if(mp.find(nums1[i])!=mp.end())
                nums1[i]=mp[nums1[i]];
            else
                nums1[i]=-1;
        }
        return nums1;
    }
};

参考资料

[1] 单调栈系列-Leetcode 496. 下一个更大元素 I