代码随想录算法训练营第四十五天| 739. 每日温度 496.下一个更大元素 I

739. 每日温度

单调栈的思路：

如果要找左边第一个比当前数字大的节点

这个栈里面存放的是index，大——》小

如果一个新节点，比栈顶大，那么就是栈顶的第一个大于它的节点

代码：

// 单调栈：一维数组，寻找 元素右边，或者左边自己大或者小的元素的位置

// 要求：找到比当前节点值高的最近节点距离
// 思路：两层for?
// 
// 思路2：单调栈 —— 如何找到比当前节点大的最新节点
// 注意：栈中存放的是Index
// 先把所有的节点按照顺序，放到从大到小的栈里，然后如果遇到新的数据，比栈顶大，那么就是栈顶的最近大顶数据
//
vector<int> dailyTemperatures(vector<int>& temperatures) {
    vector<int> result(temperatures.size(), 0);
    if (temperatures.size() == 1) return result;

    stack<int> stack_;
    stack_.push(0);

    for (int i = 1; i < temperatures.size(); i++)
    {
        // 小于等于
        if (temperatures[i] <= temperatures[stack_.top()])
        {
            stack_.push(i);
        }
        else
        {
            //如果当前节点大于，那么应该循环 直到当前节点小于等于顶点
            while (!stack_.empty() && temperatures[i] > temperatures[stack_.top()])
            {
                result[stack_.top()] = i - stack_.top() ;
                stack_.pop();
            }
            
            stack_.push(i);
        }
    }

    return result;
}

  496.下一个更大元素 I 

要求：

有一个nums1索引，根据这个索引，找到nums2的下一个更大的元素

代码：

// 根据nums1的顺序，找到对应的index，然后再找到他们是否由下一个更大的元素
// 
// 思路：先构造一个单调栈，找出来对于nums2来说，它的每一个节点对应的下一个最大节点步数
// 然后再通过nums1，进行查找，把值放进去
//
vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {
    stack<int> st;
    st.push(0);

    // map value -> index
    map<int, int> mp;

    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()])
            {
                mp.insert(make_pair(nums2[st.top()],nums2[i]));

                st.pop();
            }

            st.push(i);
        }
    }
    
    vector<int> result;
    for (int i = 0; i < nums1.size(); i++)
    {
        if (mp.find(nums1[i]) != mp.end())
        {
            result.push_back(mp[nums1[i]]);
        }
        else
        {
            result.push_back(-1);
        }
    }

    return result;
}