LeetCode-496

下一个更大元素 I

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

示例
输入: nums1 = [2,4,1], nums2 = [1,3,4,2].
输出: [-1,-1,3]
解释:
对于 num1 中的数字 2 ，第二个数组中没有下一个更大的数字，因此输出 -1 。
对于 num1 中的数字 4 ，你无法在第二个数组中找到下一个更大的数字，因此输出 -1 。
对于 num1 中的数字 1 ，第二个数组中数字1右边的下一个较大数字是 3 。

题解

方法 1 暴力

根据题意，要找出 num2 中对应元素右边第一个更大的值，可以采用暴力法求解。对于 nums1 中的每一个元素，从前向后遍历 nums2 直到找到对应元素，再继续向后遍历，若存在较大的值则返回，若不存在则返回 -1 。

方法 2 单调栈 + 哈希表

单调栈：栈内元素单调按照递增(递减)顺序排列的栈，分为单调递增栈和单调递减栈。

1. 定义
   单调递增栈： 从栈顶往栈底看，是单调递增
   单调递减栈： 从栈顶往栈底看，是单调递减
2. 性质
   a. 单调栈内的元素具有单调性
   b. 元素加入栈前，会在栈顶端删除破坏栈单调性的元素
   c. 所有元素只会进栈一次，出栈一次
   d. 使用单调栈可以找到元素向左（向右）遍历第一个比它小（大）的元素
3. 应用
   需要通过比较前后元素的大小关系来解决问题时通常使用单调栈。
   如：求解元素左边（右边）第一个更大（小）的值
4. 步骤
   假设用已知的数组元素构造单调递增栈（从前向后遍历，若存在重复元素则视题意而定），则对于每个元素：
   1）若栈为空，则直接放入该元素
   2）若栈不为空，且栈顶元素 > 当前元素，则将该元素入栈，成为新的栈顶元素
   3）若栈不为空，且栈顶元素 < 当前元素，则将栈内小于当前元素的所有元素都出栈，再将当前元素入栈

本题需要找出数组中对应元素右边第一个更大的值，可以先预处理 nums2 ,使查询对应元素右边第一个更大的值时不需要在遍历 nums2 ，因此需要先存储所有元素右边更大值的结果，再对 nums1 中每个元素进行查询得到最终结果。
找出元素右边第一个更大的值可以使用单调栈来求解，存储结果可以使用哈希表，将元素值及其右边第一个更大的元素值的对应关系存入。
单调栈求解具体过程如下：
倒序遍历 nums2 ，用单调栈维护当前位置右边的更大的元素列表，从栈底到栈顶的元素是单调递减的。对于 nums2 中的每个元素 nums2[j] ，将当前单调栈中所有小于 nums2[j] 的元素弹出单调栈，当前位置右边的第一个更大的元素即为栈顶元素，如果栈为空则说明当前位置右边没有更大的元素，然后将 nums2[j] 入栈。
注意：本题中只需要用到 nums2 中元素的相对顺序而不需要下标，故栈中直接存储 nums2 中的元素值即可。

更多题解

代码

方法 1 暴力

class Solution {
public:
    vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {
        int m=nums1.size(),n=nums2.size();
        vector<int> res(m);
        for(int i=0;i<m;i++){
            int j=0;
            //找到对应相等的元素
            //若未找到则 j=n 
            //本题中一定存在该元素
            while(j<n&&nums1[i]!=nums2[j]){
                ++j;
            }
            int k=j+1;
            //找到第一个比 nums2[j] 大的元素
            //若未找到则 k=n
            while(k<n&&nums2[k]<nums2[j]){
                ++k;
            }
            //判断是否找到符合条件的元素
            res[i]=k<n?nums2[k]:-1;
        }
        return res;
    }
};

方法 2 单调栈 + 哈希表

class Solution {
public:
    vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {
        unordered_map<int,int> hashmap;
        stack<int> st;
        int n=nums2.size();
        for(int i=n-1;i>=0;--i){
        	//弹出比当前元素小的栈内元素
            while(!st.empty()&&nums2[i]>st.top()){
                st.pop();
            }
            // hashmap 存储元素值及其右边第一个更大值的对应关系
            //若栈为空则说明当前元素右边不存在更大的值
            //若栈不为空则说明当前元素右边第一个更大的值为栈顶元素
            hashmap[nums2[i]]=st.empty()?-1:st.top();
            //将当前元素入栈
            st.push(nums2[i]);
        }
        int m=nums1.size();
        vector<int> res(m);
        for(int i=0;i<m;++i){
            res[i]=hashmap[nums1[i]];
        }
        return res;
    }
};