739.每日温度 (Medium)*

题目描述*

根据每日气温列表，请重新生成一个列表，对应位置的输入是你需要再等待多久温度才会升高超过该日的天数。如果之后都不会升高，请在该位置用 0 来代替。

例如，给定一个列表 temperatures = [73, 74, 75, 71, 69, 72, 76, 73]，你的输出应该是 [1, 1, 4, 2, 1, 1, 0, 0]。

提示：气温列表长度的范围是 [1, 30000]。每个气温的值的均为华氏度，都是在 [30, 100] 范围内的整数。

代码*

硬遍历会超时，要用栈。从头到尾扫描一遍数组，如果当天温度比栈顶温度高，就能得到结果。

栈反向

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

class Solution {
public:
    vector<int> dailyTemperatures(vector<int>& T) {
        vector <int> ans (T.size(), 0);
        stack <int> res;
        for(int i = T.size() - 1; i >= 0; --i) {
            while(!res.empty() && T[i] >= T[res.top()]) {
                res.pop();
            }
            if(!res.empty()) {
                ans[i] = res.top() - i;
            }
            res.push(i);
        }
        return ans;
    }
};

最后更新: July 23, 2022