leetcode刷题记录20-739. 每日温度

问题描述

给定一个整数数组 temperatures ，表示每天的温度，返回一个数组 answer ，其中 answer[i] 是指对于第 i 天，下一个更高温度出现在几天后。如果气温在这之后都不会升高，请在该位置用 0 来代替。

示例

示例 1:

输入: temperatures = [73,74,75,71,69,72,76,73]
输出: [1,1,4,2,1,1,0,0]

示例 2:

输入: temperatures = [30,40,50,60]
输出: [1,1,1,0]

示例 3:

输入: temperatures = [30,60,90]
输出: [1,1,0]

提示：

• 1 <= temperatures.length <= 10^5
• 30 <= temperatures[i] <= 100

问题分析：

单调栈打卡活动，每日温度。写完发现我的写法简直是浪费空间，简单说一下，我使用了栈结合pair，存了下标和温度（其实只存下标就可以了）。然后边for循环，边检查栈顶元素的温度是不是小于当前节点温度，小于就出栈并记录天数。

代码如下：

class Solution {
public:
    vector<int> dailyTemperatures(vector<int>& temperatures) {
        stack<pair<int, int>> st;
        int n = temperatures.size();
        vector<int> res(n, 0);
        for(int i = 0; i < n; ++i){
            if(st.empty()){
                st.push({i, temperatures[i]});
            }
            else{
                // 单调栈核心部分，如果栈不为空且栈顶元素小于当前节点元素，出栈记录位置
                while(!st.empty() && st.top().second < temperatures[i]){
                    res[st.top().first] = i - st.top().first;
                    st.pop();
                }
                st.push({i, temperatures[i]});
            }
        }
        return res;
    }
};