代码随想录算法训练营第五十八天【单调栈part1】 | 739. 每日温度、496.下一个更大元素 I

739. 每日温度

题目链接

力扣（LeetCode）官网 - 全球极客挚爱的技术成长平台

求解思路

其实就是找到一个元素右边第一个比自己大的元素。

单调栈的本质就是空间换时间，我们使用一个栈来记录我们遍历过的元素。

使用单调栈需要明确以下两点：

1.单调栈里存放的元素是什么？

注意本题存放的是元素的下标i，而不是元素本身。

2.单调栈的里的元素顺序是什么？（注意顺序的描述为从栈头到栈底）

如果求一个元素右边第一个更大的元素，单调栈是递增的；如果求一个元素右边第一个更小元素，单调栈是递减的。

使用单调栈主要有三个判断条件：

1.当前遍历元素T[i]小于栈顶元素T[st.top()]

将当前遍历元素T[i]进行入栈。

2.当前遍历元素T[i]等于栈顶元素T[st.top()]

将当前遍历元素T[i]进行入栈。

3.当前遍历元素T[i]大于栈顶元素T[st.top()]

将栈顶元素不断弹出，每弹出一次，更新result结果数组，直到当前遍历元素小于等于栈顶元素。

代码

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 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

题目链接

力扣（LeetCode）官网 - 全球极客挚爱的技术成长平台

求解思路

本题是单调栈套一个壳。

首先定义result结果数组，如果不存在对应位置就输出-1，因此初始化为-1。

在遍历nums2过程中，我们要判断nums2[i]是否在nums1中出现过，最后要根据nums1元素的下标来更新result数组。

注意题目中说nums1和nums2两个数组是没有重复元素，因此可以用map来做映射，可以快速找到下标，还可以判断nums2[i]是否在nums1中出现过。

单调栈的顺序（从栈头到栈底）是从小到大，即递增。

分析单调栈的三种情况：

1.当前遍历的元素T[i]小于栈顶元素T[st.top()]：

此时满足递增栈（栈头到栈底的顺序），所以直接入栈。

2.当前遍历的元素T[i]等于栈顶元素T[st.top()]

如果相等的话，依然直接入栈，因为我们要求的是右边第一个比自己大的元素，而不是大于等于。

3.当前遍历的元素T[i]大于栈顶元素T[st.top()]

此时找到右边第一个比自己大的元素，判断栈顶元素是否在nums1里出现过，（注意栈里的元素是nums2的元素），如果出现过，开始记录结果。

代码

class Solution {
public:
    vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {
        stack<int> st;
        vector<int> result(nums1.size(), -1);
        if (nums1.size() == 0) return result;

        unordered_map<int, int> umap;
        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 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){ // 看map里是否存在这个元素
                        int index = umap[nums2[st.top()]]; // 根据map找到nums2[st.top()] 在 nums1中的下标
                        result[index] = nums2[i];
                    }
                    st.pop();
                }
                st.push(i);
            }
        }
        return result;
    }
};