496. 下一个更大元素 I

题目链接

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

题目描述

题目分析

两个 没有重复元素 的数组 nums1 和 nums2，所以可以用map做映射。
先找nums2的“最大”，找到后再映射(如果nums1中存在)到nums1。
注意：res中找的不是索引，是值。

class Solution {
public:
    vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {
        int n1 = nums1.size();
        int n2 = nums2.size();

        vector<int> res(n1, -1);
        if(n1 == 0) return res;

        stack<int> st;
        st.push(0);
        
        unordered_map<int, int> umap;
        for(int i = 0; i < n1; i++){
            umap[nums1[i]] = i;
        }
        for(int i = 1; i < n2; i++){
            while(!st.empty() && nums2[i] > nums2[st.top()]){//
                if(umap.count(nums2[st.top()]) > 0){//nums2的值有在nums1中的
                    int index = umap[nums2[st.top()]];
                    res[index] = nums2[i];
                }
                st.pop();
            }
            st.push(i);
        }
        return res;
    }
};