算法训练第五十八天

总结:今日事单调栈的开端,还是挺巧妙的。

496. 下一个更大元素 I - 力扣(LeetCode)

代码:

class Solution {
public:
    vector nextGreaterElement(vector& nums1, vector& nums2) {
        vector res(nums1.size(),-1);
        unordered_map umap;

        if(nums1.size() == 0)
        return res;
        for(int i = 0; i < nums1.size();i++)
        umap[nums1[i]] = i;

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

739. 每日温度 - 力扣(LeetCode)

class Solution {
public:
    vector dailyTemperatures(vector& temperatures) {
        stack st;
        vector 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;
    }
};

你可能感兴趣的:(算法)