leetcode 496. 下一个更大元素 I-Toy模板网

这篇具有很好参考价值的文章主要介绍了leetcode 496. 下一个更大元素 I。希望对大家有所帮助。如果存在错误或未考虑完全的地方，请大家不吝赐教，您也可以点击"举报违法"按钮提交疑问。

2023.8.28

leetcode 496. 下一个更大元素 I,leetcode专栏,leetcode,算法,职场和发展,c++,数据结构

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

暴力解：

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查询相应元素对应的查询值。代码如下：文章来源地址https://www.toymoban.com/news/detail-680537.html

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;
    }
};

到了这里，关于leetcode 496. 下一个更大元素 I的文章就介绍完了。如果您还想了解更多内容，请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章，希望大家以后多多支持TOY模板网！