LeetCode刷题（161）～下一个更大元素 I哈希+单调栈

Posted 2023-01-14

题目描述

给定两个 没有重复元素 的数组 nums1 和 nums2 ，其中nums1 是 nums2 的子集。找到 nums1 中每个元素在 nums2 中的下一个比其大的值。

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

示例 1:

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

示例 2:

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

提示：

• nums1和nums2中所有元素是唯一的。
• nums1和nums2 的数组大小都不超过1000。

解答 By 海轰

提交代码（暴力）

vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) 
        vector<int> ans;
        for(int num:nums1)
        
            for(int i=0;i<nums2.size();++i)
            
                if(nums2[i]==num)
                
                    int j;
                    for(j=i;j<nums2.size();++j)
                    
                        if(nums2[j]>num)
                        
                            ans.push_back(nums2[j]);
                            break;
                        
                    
                    if(j==nums2.size())
                    ans.push_back(-1);
                
            
        
        return ans;
    

运行结果

提交代码（暴力 优化）

vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) 
       vector<int> ans;
       unordered_map<int,int> m;// 利用哈希记录每个元素的下标
       for(int i=0;i<nums2.size();++i)
       m[nums2[i]]=i;
       for(int num:nums1)
       
           int index=m[num];
           int j;
           for(j=index;j<nums2.size();++j)
           
               if(nums2[j]>num)
               
                   ans.push_back(nums2[j]);
                   break;
               
           
           if(j==nums2.size())
           ans.push_back(-1);
       
       return ans;
    

运行结果

提交代码（哈希+栈）

vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) 
       unordered_map<int,int> m;
       stack<int> s;
       for(int i=0;i<nums2.size();++i)
       
           if(s.empty())
                s.push(nums2[i]);
           else
           
               if(nums2[i]>s.top())
               
                   m[s.top()]=nums2[i];
                   s.pop();
                   --i;
               
               else
               
                   s.push(nums2[i]);
               
           
       
       while(!s.empty())
       
           m[s.top()]=-1;
           s.pop();
       
       vector<int> ans;
       for(int num:nums1)
       
           ans.push_back(m[num]);
       
       return ans;
    

运行结果

提交代码（哈希+栈 简化）

vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) 
       unordered_map<int,int> m;
       stack<int> s;
       for(int i=0;i<nums2.size();++i)
       
           if(!s.empty()&&nums2[i]>s.top())
            
                   m[s.top()]=nums2[i];
                   s.pop();
                   --i;
           
           else
           
                   s.push(nums2[i]);
           
       
       while(!s.empty())
       
           m[s.top()]=-1;
           s.pop();
       
       vector<int> ans;
       for(int num:nums1)
       
           ans.push_back(m[num]);
       
       return ans;
    

运行结果

题目来源

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/next-greater-element-i