算法训练第五十八天

最新推荐文章于 2024-09-02 16:35:28 发布

zhangke_EX

最新推荐文章于 2024-09-02 16:35:28 发布

阅读量3.8k

点赞数

文章标签：算法

本文链接：https://blog.csdn.net/zhangke_EX/article/details/132746226

版权

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

496. 下一个更大元素 I - 力扣（LeetCode）

代码：

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

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

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