代码随想录二刷-下一个更大元素 I

496：下一个更大元素 I

链接: Leetcode496.下一个更大元素 I
Leetcode Hot100二刷-每日温度

题目需求

nums1 中数字 x 的 下一个更大元素 是指：x 在 nums2 中对应位置右侧的 第一个 比 x 大的元素。

给你两个没有重复元素的数组 nums1 和 nums2 ，下标从 0 开始计数，其中nums1 是 nums2 的子集。
对于每个 $0\le i<nums1.length$，找出满足 nums1[i] == nums2[j] 的下标 j ，并且在 nums2 确定 nums2[j] 的 下一个更大元素 。如果不存在下一个更大元素，那么本次查询的答案是 -1 。

返回一个长度为 nums1.length 的数组 ans 作为答案，满足 ans[i] 是如上所述的 下一个更大元素

示例：
输入：nums1 = [4,1,2], nums2 = [1,3,4,2].
输出：[-1,3,-1]
解释：nums1 中每个值的下一个更大元素如下所述：

• 4 ，用加粗斜体标识，nums2 = [1,3,4,2]。不存在下一个更大元素，所以答案是 -1 。
• 1 ，用加粗斜体标识，nums2 = [1,3,4,2]。下一个更大元素是 3 。
• 2 ，用加粗斜体标识，nums2 = [1,3,4,2]。不存在下一个更大元素，所以答案是 -1 。

思路

1. 分析题干
   a. 题目求的是nums1每个元素在nums2里的下一个更大元素。所以划分为两个步骤，首先是找到该元素在nums2里的位置，其次是看下一个更大的元素是否存在【第一步】
   b. 既然是看nums2里各元素的下一个更大元素，那么可以先按照每日温度的思路把nums2里的所有元素的下一个更大元素记录在一个vector容器里，记为vec【第二步】
   c. 接着，把nums2的元素用哈希表存好【第三步】
   d. 最后遍历nums1元素，当前元素在nums2里的位置去哈希表里找，找到后，再看vec对应位置的元素【第四步】

2. 写题顺序
   a. 用单调栈记录nums2里所有元素的下一个更大元素
   b. 用哈希表存nums2里所有元素
   c. 遍历nums1数组，根据第四步的步骤收获结果

3. 注意事项
   a. 初始化数组的时候，记得初始化为-1

4. 代码

class Solution {
public:
    vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {
        if(nums2.size() == 1) return vector<int>{-1};
        int n = nums2.size();
        vector<int> vec(n, -1);
        stack<int> sta;
        sta.push(0);
        for(int i = 1; i < nums2.size(); ++i){
            while(!sta.empty() && nums2[i] > nums2[sta.top()]){
                vec[sta.top()] = nums2[i];
                sta.pop();
            }
            sta.push(i);
        }

        unordered_map<int, int> map;
        for(int i = 0; i < n; ++i){
            map[nums2[i]] = i;
        }

        vector<int> result(nums1.size(), -1);
        for(int i = 0; i < nums1.size(); ++i){
            int idx = map[nums1[i]];
            result[i] = vec[idx];
        }

        return result;
    }
};

参考: 代码随想录-496.下一个更大元素 I