LeetCode 496下一个更大元素I

LeetCode 496下一个更大元素I^Easy

• 题目简述：给定两个 没有重复元素 的数组 nums1 和 nums2 ，其中nums1 是 nums2 的子集。找到 nums1 中每个元素在 nums2 中的下一个比其大的值。nums1 中数字 x 的下一个更大元素是指 x 在 nums2 中对应位置的右边的第一个比 x 大的元素。如果不存在，对应位置输出 -1 。

• 输入：nums1 = [4,1,2], nums2 = [1,3,4,2]. 输出：[-1,3,-1]

  输入：nums1 = [2,4], nums2 = [1,2,3,4]. 输出：[3,-1]

• 思路：哈希表+单调栈

  • 首先遍历nums2中的所有元素，利用单调栈找出其中每一个当前元素与下一个更大元素的对应关系，用哈希表记录该关系
  • 然后遍历nums1中所有元素，如果当前元素在哈希表中出现，那么其对应的键值就是答案

class Solution {
public:
    vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {
        vector<int> res(nums1.size(), -1);
        unordered_map<int, int> hash;
        stack<int> st;

        for(auto &num : nums2)
        {
            while(!st.empty() && num > st.top())
            {
                hash[st.top()] = num;
                st.pop();
            }
            st.push(num);
        }

        for(int i = 0; i < nums1.size(); i++)
        {
            if(hash.count(nums1[i]))
                res[i] = hash[nums1[i]];
        }

        return res;
    }
};