leetcode-496. 下一个更大元素 I刷题笔记（c++）

一时失言乱红尘 2023-02-22 05:03 12阅读 0赞

#### 写在前面 ####

*  难度：简单
 *  `双层循环` 或 `堆栈 + 数据映射`
    
     *  后者理解、尝试有1定难度

#### 题目详情 ####

给定两个 没有重复元素 的数组 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 = [1,3,5,2,4], nums2 = [6,5,4,3,2,1,7].
    输出: [7,7,7,7,7]
    
    提示：
    
    nums1和nums2中所有元素是唯一的。
    nums1和nums2 的数组大小都不超过1000。

#### ac代码 ####

*  `2层循环`
    
     *  `vector<int>`元素查找
        
         *  `vector<int>::iterator it = find(nums2.begin(), nums2.end(), c);`

class Solution
    {
        
    public:
        vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2)
        {
        
            vector<int> ans;
            for(auto c : nums1)
            {
        
                vector<int>::iterator it = find(nums2.begin(), nums2.end(), c);
                for(it=it+1; it != nums2.end(); ++it)
                {
        
                    if(*it > c)
                    {
        
                        ans.push_back(*it);
                        break;
                    }
                }
                if(*(it-1) <= c && it == nums2.end())
                    ans.push_back(-1);
            }
            return ans;
        }
    };

*  `堆栈 + 数据映射`
    
     *  简化后的个人逻辑
     *  解题思想
        
         *  遍历nums2数组，逐个元素入栈；
         *  当前元素大于栈顶元素，栈顶元素出栈；并将栈顶元素与当前元素封装为key-value键值对存入`unordered_map<int, int> umap`；
         *  针对堆栈元素循环至条件不成立，得到nums2中元素所对应的下1个更大元素的Map表；
         *  遍历nums1，`umap`中存在则输出，不存在输出-1。

class Solution
    {
        
    public:
    	vector<int> nextGreaterElement(vector<int> &nums1, vector<int> &nums)
    	{
        
    	    vector<int> ans;
    	    if(nums.size() <=0)
    	        return ans;
    	    stack<int> stk;
    	    unordered_map<int, int> umap;
    	    for (int i = 0; i < nums.size(); ++i)
    	    {
        
    	        while (stk.size() && stk.top() < nums[i])
    	        {
        
    	            umap[stk.top()] = nums[i];
    	            stk.pop();
    	        }
    	        stk.push(nums[i]);
    	    }
    	    for (int num: nums1)
    	    {
        
    	        if(umap.count(num))
    	            ans.push_back(umap[num]);
    	        else
    	            ans.push_back(-1);
    	    }
    	    return ans;
    	}
    };

*  参考文章
    
     *  [Leetcode刷题52-496. 下一个更大元素 I(C++详细解法！！！)][Leetcode_52-496. _ I_C]

[Leetcode_52-496. _ I_C]: https://blog.csdn.net/qq_40858438/article/details/89203277