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

739. 每日温度

题目链接：739. 每日温度

文档讲解：代码随想录

视频讲解：单调栈，你该了解的，这里都讲了！LeetCode:739.每日温度

思路

求右侧第一个比自己大的元素的下标，使用单调栈存放元素下标，从栈头到栈尾递增。

如果当前元素小于等于栈顶元素，则讲当前元素入栈；

如果当前元素大于栈顶元素，该元素即为栈顶元素右侧第一个比自己大的元素的下标。

代码

class Solution {
public:
    vector<int> dailyTemperatures(vector<int> &temperatures) {
        vector<int> result(temperatures.size(), 0);

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

496.下一个更大元素 I

题目链接：496.下一个更大元素 I

文档讲解：代码随想录

视频讲解：单调栈，套上一个壳子就有点绕了| LeetCode:496.下一个更大元素

思路

使用单调栈遍历nums2的元素，使用map哈希表映射nums1中的元素和下标。

如果nums2的当前元素小于等于栈顶元素，将该元素入栈；

如果nums2的当前元素大于栈顶元素，则判断栈顶元素是否在nums1中出现过，如果出现则记录结果

代码

class Solution {
public:
    vector<int> nextGreaterElement(vector<int> &nums1, vector<int> &nums2) {
        vector<int> result(nums1.size(), -1);
        stack<int> st;
        unordered_map<int, int> umap; // key: 下标元素, value: 下标
        for (int i = 0; i < nums1.size(); i++)
            umap[nums1[i]] = i;

        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.find(nums2[st.top()]) != umap.end())
                        result[umap[nums2[st.top()]]] = nums2[i];
                    st.pop();
                }
                st.push(i);
            }
        }
        return result;
    }
};