代码随想录|day58|单调栈part01 ● 739. 每日温度 ● 496.下一个更大元素 I

739. 每日温度

链接:代码随想录

今天正式开始单调栈,这是单调栈一篇扫盲题目,也是经典题。

大家可以读题,思考暴力的解法,然后在看单调栈的解法。 就能感受出单调栈的巧妙

代码随想录|day58|单调栈part01 ● 739. 每日温度 ● 496.下一个更大元素 I_第1张图片

fvfdvsf

第一遍写错:

class Solution {
public:
    vector dailyTemperatures(vector& temperatures) {
        int n=temperatures.size();
        vectorres;
        if(n==1)
        {
          res.push_back(0);
          return res;
        }
        stackst;
        st.push(0);
        for(int i=1;itemperatures[st.top()])
                {
                    res.push_back(i-st.top());
                    st.pop();
                }
                st.push(i);
            }

        }
        return res;

        
    }
};

主要出错点分析:

代码随想录|day58|单调栈part01 ● 739. 每日温度 ● 496.下一个更大元素 I_第2张图片

 按照这个代码去写,因为栈pop出来是逆序的,所以res 加入的index也是部分逆序的,不符合题目要求。

正确写法

class Solution {
public:
    vector dailyTemperatures(vector& temperatures) {
        int n=temperatures.size();
        vectorres(n,0);
        if(n==1)
        {
          return res;
        }
        stackst;
        st.push(0);
        for(int i=1;itemperatures[st.top()])
                {
                    res[st.top()]=i-st.top();
                    st.pop();
                }
                st.push(i);
            }

        }
        return res;

        
    }
};

496.下一个更大元素 I

链接:代码随想录

本题和 739. 每日温度 看似差不多,其实 有加了点难度。

代码随想录|day58|单调栈part01 ● 739. 每日温度 ● 496.下一个更大元素 I_第3张图片

代码随想录|day58|单调栈part01 ● 739. 每日温度 ● 496.下一个更大元素 I_第4张图片

这个整个的逻辑有点绕,到底要不要创建map?map的定义装什么?为了便于理解,我定义了map,遍历nums2中的单调栈,map的第一个元素是nums2中的元素,第二个元素储存nums2中每个元素对应的下一个元素。然后最终修改nums1.看nums1中的每个元素在不在map中,从而找到对应的下一个元素。

class Solution {
    /*
    如果计算nums2中所有元素的下一个元素,会造成时间和空间的浪费,但是又必须计算
    */
public:
    vector nextGreaterElement(vector& nums1, vector& nums2) {
        int n1=nums1.size();
        int n2=nums2.size();
        unordered_mapum;//存储nums2中所有的下一个元素
        stacks;
        s.push(nums2[0]);

        for(int i=1;is.top())
                {
                    um.insert(make_pair(s.top(),nums2[i]));
                    s.pop();
                }
                s.push(nums2[i]);
            }
        }

        //打印看一下um
        for(auto p :um)
        {
            cout<

503.下一个更大元素II 

链接:代码随想录

因为太好奇了忍不住提前做了。怎样层层递进加入难度的。

代码随想录|day58|单调栈part01 ● 739. 每日温度 ● 496.下一个更大元素 I_第5张图片

代码随想录|day58|单调栈part01 ● 739. 每日温度 ● 496.下一个更大元素 I_第6张图片 我记得的,这个要把数组*2,或者使用%。以前做过,现在不记得咋做的了。

如果不存在循环着找,那这道题的答案是:

class Solution {
public: 
//因为不涉及index,可以直接把数组扩展为原来的两倍,要么考虑(i+k)%n
    vector nextGreaterElements(vector& nums) {
        int n=nums.size();
        stacks;
        s.push(nums[0]);
        vectorres(n,-1);//储存下一个最大元素
        for(int i=1;is.top())
                {
                     res[i-1]=nums[i];
                     s.pop();
                }
                s.push(nums[i]);
            }
        }
        /*for(auto p : res)
        {
            cout<

打印结果:

代码随想录|day58|单调栈part01 ● 739. 每日温度 ● 496.下一个更大元素 I_第7张图片

 注意代码:

最后的代码: 理解起来还是比较抽象:

class Solution {
public:
//一般是通过 % 运算符求模(余数),来模拟环形特效
/*
所谓环形数组:
利用%n进行
while(1)
{
    cout< nextGreaterElements(vector& nums) {
        int n=nums.size();
        /*这道题基本最边界的情况:最后一个元素找下一个最大元素,结果在它的左边找到了。所以基本只要
    遍历两遍,就能得到答案
*/
        stackst;
        vectorres(n,-1);
        st.push(0);
        for(int i=1;i<2*n;i++)
        {
            int index=i%n;
            if(nums[index]<=nums[st.top()])
            {
                st.push(index);
            }
            else
            {
                while(!st.empty()&& nums[index]>nums[st.top()])
                {
                      res[st.top()]=nums[index];
                      st.pop();
                }
                st.push(index);
            }

        }
        return res;
    }
};

你可能感兴趣的:(代码随想录,数据结构)