力扣 496 下一个更大的元素I

题目

496. 下一个更大元素 I

思路

接到手，肯定是考虑直接暴力模拟的，时间复杂度是O(mn)。题解给了一个更好的方法，利用单调栈来对nums2进行预处理。这个方法实在是很妙。大致的思路是，倒序遍历nums2，对于某个循环到的数num来说，如果num比这个栈顶元素大，那么这个元素出栈，因为它不可能是num右边第一个比它大的数。
对于这个循环结束有两种情况，第一是栈顶元素大于num，那么这个栈顶元素就是这个num所对应的数；第二是栈空了，说明右边没有满足题意的数，对应-1。
注意最后需要把num入栈。

AC代码

class Solution {
public:
    vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {
        unordered_map<int,int> m;
        stack<int> s;
        int len=nums2.size(),len1=nums1.size();
        for(int i=len-1;i>=0;--i){
            int num=nums2[i];
            while(!s.empty()&&num>s.top()){
                s.pop();
            }
            m[num]=(s.empty())? -1:s.top();
            s.push(num);
        }
        vector<int> res;
        for(int i=0;i<len1;++i){
            res.push_back(m[nums1[i]]);
        }
        return res;
    }
};

后记

最近可能是受了啥刺激emm，感觉对学习又提起了热情。