496. 下一个更大元素 I

题目描述

给你两个 没有重复元素 的数组 nums1 和 nums2 ，其中nums1 是 nums2 的子集。

请你找出 nums1 中每个元素在 nums2 中的下一个比其大的值。

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/next-greater-element-i
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

题目分析

我们首先考虑，对于num1的每次查询，我们都要遍历一遍num2数组找出num2数组的第i位的右边第一个比它大的数是啥，所以此处我们不妨先将num2进行一次预处理，计算num2数组每一位的右边第一个比它大的是什么。

所以目前问题就变成了计算num2数组每一位的右边第一个比它大的是什么，倒序遍历 ，并用单调栈中维护当前位置右边的更大的元素列表，从栈底到栈顶的元素是单调递减的，又因为题目中说不存在重复元素，因此我们使用一个hashmap进行存储，对每一位进行维护即hashmap[temp]=st.empty() ? -1 : st.top(); temp为当前指向的num2的值，st为单调栈。假如栈空，说明没有比它大的，则为-1 。

最后，针对num1中每位查询hashmap即可。

参考代码

class Solution {
public:
    vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {
        map<int,int> hashmap;
        stack<int> st;
        vector<int> ans;
        for(int i=nums2.size()-1;i>=0;i--){
            int temp=nums2[i];
            while(!st.empty() && st.top()<=temp){
                st.pop();
            }
            hashmap[temp]=st.empty() ? -1 : st.top();
            st.push(temp);
        }
        for(auto i:nums1){
            ans.push_back(hashmap[i]);
        }
        return ans;
    }
};