[LeetCode] 739、每日温度

题目描述

根据每日 气温 列表，请重新生成一个列表，对应位置的输出是你需要再等待多久温度才会升高超过该日的天数。如果之后都不会升高，请在该位置用 0来代替。例如，给定一个列表 temperatures = [23, 24, 25, 21, 19, 22, 26, 23]，你的输出应该是 [1, 1, 4, 2, 1, 1, 0, 0]。

解题思路

想到过用“单调栈”去实现，但是在模拟过程中因为没有搞清输出顺序，导致失败。（思路要灵活）

• 暴力法：是针对每个温度值向后进行依次搜索，找到比当前温度更高的值，这样的计算复杂度就是 $O(n^2)$。
• 单调栈（存下标，能很快计算天数）：此处需要维护一个“单调递减栈”。（在输出的时候其实直接输出到已经初始化好的vector里面去就好了，因为之前存的是下标所以很方便），看题解动图。

参考代码

class Solution {
public:
    vector<int> dailyTemperatures(vector<int>& T) {
        int length = T.size();
        if(length == 0)
            return vector<int>();
        
        vector<int> days(length, 0);
        stack<int> mStack;  // 存下标，便于计算天数
        for(int i = 0; i < length; i++){
            while(!mStack.empty() && T[i] > T[mStack.top()]){
                days[mStack.top()] = i - mStack.top();
                mStack.pop();
            }
            mStack.push(i);
        }
        
        return days;
    }
};