【LeetCode】739. 每日温度

739. 每日温度（中等）

思路

• 我们可以维持一个单调递减的栈，表示每天的温度；为了方便计算天数差，这里存放位置（下标），而非温度本身。因为温度可以通过访问数组下标获取。

• 从左向右遍历数组，对于每个日期 p ，如果 p 的温度比栈顶存储位置 q 的温度高，则我们取出 q，并记录 q 需要等待的天数，即 q-p ；重复这一过程，直到 p 的温度小于等于栈顶元素位置的温度（或者空栈）时，将 p 插入栈顶，然后考虑下一天。在这个过程中，栈内元素永远保持单调递减。

• 最后如果栈内剩余一些日期，则说明它们之后都没有出现更暖和的日期，置为 0。

代码

class Solution {
public:
    vector<int> dailyTemperatures(vector<int>& temperatures) {
        int n = temperatures.size();
        vector<int> ans(n);
        stack<int> st;
        for(int i=0; i<n; ++i){
            while(!st.empty() && temperatures[i] > temperatures[st.top()]){
                ans[st.top()] = i - st.top();
                st.pop();
            }
            st.push(i);
        }
        return ans;
    }
};

收获

• 想到了单调栈的做法，但是我只想到了把温度存入栈，没想到可以存入下标。