代码随想录算法训练营第五十天|leetcode496、739题

最新推荐文章于 2025-07-02 15:11:11 发布

原创最新推荐文章于 2025-07-02 15:11:11 发布 · 327 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#算法

本文介绍了两个LeetCode题目，分别使用了栈和哈希表的数据结构来解决：第739题利用栈找出每天的温度上升天数，第496题通过栈和哈希表找到nums1中每个元素在nums2中的下一个更大值。

一、leetcode第739题

class Solution {
public:
    vector<int> dailyTemperatures(vector<int>& temperatures) {
    stack<int>st;
    vector<int>result(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()])
            {
                result[st.top()]=i-st.top();
                st.pop();
            }
            st.push(i);
        }
    }
    return result;
    }
};

二、leetcode第496题

class Solution {
public:
    vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {
    vector<int>result(nums1.size(),-1);
    unordered_map<int,int>unmap;
    stack<int>st;
    for(int i=0;i<nums1.size();i++)
    {
        unmap[nums1[i]]=i;
    }
    if(nums1.size()==0)
    {
        return result;
    }
    st.push(0);
    for(int i=1;i<nums2.size();i++)
    {
        if(nums2[i]<=nums2[st.top()])
        {
            st.push(nums2[i]);
        }
        else
        {
            while(!st.empty()&&nums2[i]>nums2[st.top()])
            {
                if(unmap.count(nums2[st.top()])>0)
                {
                    int index=unmap[nums2[st.top()]];
                    result[index]=nums2[i];
                }
                st.pop();
            }
            st.push(i);
        }
    }
    return result;
    }
};