力扣 496. 下一个更大元素 I 单调栈 哈希

https://leetcode-cn.com/problems/next-greater-element-i/
力扣 496. 下一个更大元素 I 单调栈 哈希_第1张图片
思路:题目说的花里胡哨……其实就是对于 n u m s 2 nums2 nums2的每一个元素,求出位于其右侧且大于它的一个元素的值就可以了,搞一个单调栈就行了,维护一个单调递减栈即可在线性复杂度内求出。但是考虑到 n u m s 1 nums1 nums1是乱序的,所以需要再搞一个哈希表。

class Solution {
     
public:
    vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {
     
        //维护一个单调非升栈
        stack<int> s;
        unordered_map<int,int> m;
        int siz1=nums1.size(),siz2=nums2.size();
        for(int i=0;i<siz2;i++){
     
            while(!s.empty()&&nums2[i]>s.top())
                m[s.top()]=nums2[i],s.pop();
            s.push(nums2[i]);
        }
        while(!s.empty())
            m[s.top()]=-1,s.pop();
        vector<int> ans(siz1);
        for(int i=0;i<siz1;i++)
            ans[i]=m[nums1[i]];
        return ans;
    }
};

你可能感兴趣的:(面试题,单调栈,哈希)