【代码随想录算法训练Day53】LeetCode 739.每日温度、LeetCode 496.下一个更大元素、LeetCode 503. 下一个更大元素 II

最新推荐文章于 2024-07-24 22:13:06 发布

Frostnova丶

最新推荐文章于 2024-07-24 22:13:06 发布

阅读量189

点赞数 3

分类专栏：代码随想录文章标签：算法 leetcode 单调栈

本文链接：https://blog.csdn.net/qq_46121420/article/details/140083365

版权

代码随想录专栏收录该内容

58 篇文章 0 订阅

订阅专栏

Day53 单调栈

LeetCode 739.每日温度

经典的单调栈题目，确实的感受到了单调栈的强大之处。

class Solution {
public:
    vector<int> dailyTemperatures(vector<int>& temperatures) {
        stack<int> st;
        vector<int> res(temperatures.size(),0);
        st.push(0);
        for(int i=1;i<temperatures.size();i++){
            if(temperatures[i]<=temperatures[st.top()])
                st.push(i);
            else{
                while(!st.empty() && temperatures[i]>temperatures[st.top()]){
                    res[st.top()]=i-st.top();
                    st.pop();
                }
                st.push(i);
            }
        }
        return res;
    }
};

LeetCode 496.下一个更大元素I

这道题和前一道题相比多绕了一个弯，使用了两个数组，但就是这一个弯我们就要多考虑很多东西。我们要先用哈希表存储值的映射，用数组2里的数值与它在数组1的下标相映射，然后就是下一个最大元素的做法了。

class Solution {
public:
    vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {
        stack<int> st;
        vector<int> res(nums1.size(),-1);
        if(nums1.size()==0) return res;

        unordered_map<int,int> umap;
        for(int i=0;i<nums1.size();i++)
            umap[nums1[i]]=i;
        st.push(0);
        for(int i=1;i<nums2.size();i++){
            if(nums2[i]<=nums2[st.top()])
                st.push(i);
            else{
                while(!st.empty() && nums2[i]>nums2[st.top()]){
                    if(umap.count(nums2[st.top()])>0){
                        int index=umap[nums2[st.top()]];
                        res[index]=nums2[i];
                    }
                    st.pop();
                }
                st.push(i);
            }
        }
        return res;
    }
};

LeetCode 503. 下一个更大元素 II

这道题又是数组成环题，我们的处理方法是，让i走两倍数组长度，但是为了不越界，在进行下标操作时全部是用i%nums.size()来操作，这样就在形式上完成了转两圈的遍历。

class Solution {
public:
    vector<int> nextGreaterElements(vector<int>& nums) {
        vector<int> res(nums.size(),-1);
        if(nums.size()==0) return res;

        stack<int> st;
        st.push(0);
        for(int i=1;i<nums.size()*2;i++){
            if(nums[i%nums.size()]<=nums[st.top()])
                st.push(i%nums.size());
            else{
                while(!st.empty() && nums[i%nums.size()]>nums[st.top()]){
                    res[st.top()]=nums[i%nums.size()];
                    st.pop();
                }
                st.push(i%nums.size());
            }
        }
        return res;
    }
};