leetcode 496. 下一个更大元素 I

最新推荐文章于 2024-09-27 15:00:48 发布

我真的很帅阿

最新推荐文章于 2024-09-27 15:00:48 发布

阅读量266

点赞数

分类专栏： leetcode刷题记录文章标签： leetcode 算法职场和发展 c++ 数据结构

本文链接：https://blog.csdn.net/m0_61028090/article/details/132538281

版权

leetcode刷题记录专栏收录该内容

206 篇文章 2 订阅

订阅专栏

本文介绍了在没有重复元素的数组中，使用暴力解法和单调栈算法求解nextGreaterElement问题的实现。暴力解法逐个查找，而单调栈法利用哈希map和栈结构优化查询过程。

摘要由CSDN通过智能技术生成

2023.8.28

这题提供暴力解法和单调栈法两种方法。

暴力解：

class Solution {
public:
    vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {
        vector<int> ans(nums1.size(),-1);
        for(int i=0; i<nums1.size(); i++)
        {
            int j=0;
            while(j<nums2.size() && nums2[j]!=nums1[i]) j++;
            j++;
            while(j<nums2.size() && nums2[j]<=nums1[i]) j++;
            if(j!=nums2.size()) ans[i] = nums2[j];
        }
        return ans;
    }
};

单调栈：

题中说了数组中没有重复的元素，为了避免重复遍历nums2数组，可以用一个哈希map和单调栈stack将nums2数组中的每个元素对应其查询值存储在map中，类似于每日温度。然后再遍历nums1查询相应元素对应的查询值。代码如下：

class Solution {
public:
    vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {
        unordered_map<int,int> hashmap;
        stack<int> stk;
        for(int i=nums2.size()-1; i>=0; i--)
        {
            while(!stk.empty() && nums2[i]>=stk.top()) stk.pop();
            hashmap[nums2[i]] = stk.empty()? -1 : stk.top(); //栈为空说明没有比当前元素更大的元素了，即查询结果为-1。
            stk.push(nums2[i]);
        }
        vector<int> ans(nums1.size());
        for(int i=0; i<nums1.size(); i++)
        {
            ans[i] = hashmap[nums1[i]];
        }
        return ans;
    }
};