LeetCode 496. 下一个更大元素 I、503. 下一个更大元素 II、739. 每日温度（单调栈1）

17231

已于 2022-06-02 19:02:53 修改

阅读量68

点赞数

分类专栏： LeetCode 文章标签： leetcode 算法

于 2022-06-02 17:36:35 首次发布

本文链接：https://blog.csdn.net/qq_50117046/article/details/125104576

版权

LeetCode 专栏收录该内容

10 篇文章 0 订阅

订阅专栏

这篇博客介绍了如何使用单调栈解决寻找数组中下一个更大元素的问题。在496题和503题中，通过建立单调递减栈，可以高效地找到每个元素的下一个更大值。739题中，单调栈用于计算每日温度中需要等待多少天才能达到更高温度。这些题目展示了单调栈在处理序列数据时的有效性。

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

总结

单调栈作用：在一维数组中找第一个满足某种条件的数
更大元素==>递减栈

496. 下一个更大元素 I

题目

nums1 中数字 x 的下一个更大元素是指 x 在 nums2 中对应位置右侧的第一个比 x 大的元素。

给你两个没有重复元素的数组 nums1 和 nums2 ，下标从 0 开始计数，其中nums1 是 nums2 的子集。

对于每个 0 <= i < nums1.length ，找出满足 nums1[i] == nums2[j] 的下标 j ，并且在 nums2 确定 nums2[j] 的下一个更大元素。如果不存在下一个更大元素，那么本次查询的答案是 -1 。

返回一个长度为 nums1.length 的数组 ans 作为答案，满足 ans[i] 是如上所述的下一个更大元素。

思路

这里使用unordered_map因为题目中的nums1是nums2的子集，且nums1和nums2中所有整数互不相同。

class Solution {
public:
    vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {
        stack<int> st;
        //key-value表示nums[i]的下一个更大元素nums[j]
        unordered_map<int,int> hm;
        for(int i=0;i<nums2.size();i++){
            while(!st.empty()&&nums2[i]>nums2[st.top()]){
                int tmp=st.top();
                st.pop();
                hm[nums2[tmp]]=nums2[i];
            }
            st.push(i);
        }
        while(!st.empty()){
            int tmp=st.top();
            st.pop();
            hm[nums2[tmp]]=-1;
        }
        vector<int> ans;
        for(int i=0;i<nums1.size();i++){
            ans.push_back(hm[nums1[i]]);
        };
        return ans;
    }
};

503. 下一个更大元素 II

直接使用ans记录结果，无需使用unordered_map

class Solution {
public:
    vector<int> nextGreaterElements(vector<int>& nums) {
        int n=nums.size();
        vector<int> ans(n,-1);
        stack<int> st;
        for(int i=0;i<2*n-1;i++){
            while(!st.empty()&&nums[i%n]>nums[st.top()]){
                int tmp=st.top();
                st.pop();
                ans[tmp]=nums[i%n];
            }
            st.push(i%n);
        }
        return ans;
    }
};

739. 每日温度

class Solution {
public:
    vector<int> dailyTemperatures(vector<int>& temperatures) {
        int n=temperatures.size();
        vector<int> ans(n,0);
        stack<int> st;
        for(int i=0;i<n;i++){
            while(!st.empty()&&temperatures[i]>temperatures[st.top()]){
                int tmp=st.top();
                st.pop();
                //记录结果为天数之差
                ans[tmp]=i-tmp;
            }
            st.push(i);
        }
        return ans;
    }
};