leetcode 739. 每日温度

leetcode 739. 每日温度

题目详情

题目链接
请根据每日 气温 列表，重新生成一个列表。对应位置的输出为：要想观测到更高的气温，至少需要等待的天数。如果气温在这之后都不会升高，请在该位置用 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) {
        int len = T.size();
        vector<int> res(len, 0);
        int end = len - 1;
        for (int i = len - 1; i >= 0; --i) {
            if (T[i] >= T[end]) {
                res[i] = 0, end = i;
            } else {
                if (T[i] < T[i + 1]) {
                    res[i] = 1;
                } else {
                    int loc = i + 1 + res[i + 1];
                    while (T[i] >= T[loc]) {
                        loc += res[loc];
                    }
                    res[i] = loc - i;
                }
            }
        }
        return res;
    }
};

我的成绩

执行结果：通过
执行用时 :104 ms, 在所有 C++ 提交中击败了80.13%的用户
内存消耗 : 34.3 MB, 在所有 C++ 提交中击败了7.69%的用户

一些想法

本道题我的想法首先是暴力，但暴力效率太低，原因是每次的搜索范围太大，所以可以考虑从数组最后开始查找，end的位置是从当前位置开始到数组最后的最大值所在的位置，则每个结果的产生是在当前位置与end之间，然后我们可以通过循环找到每个位置所对应的结果来查找。

执行用时为 28 ms 的范例

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

思考

参见官方解答