【栈 - LeetCode】739.每日温度

题解

暴力+技巧

官网给的一个暴力遍历的方式，技巧点在于，温度的最大值是 100，因此里面的 for 循环可以通过控制最大是到 100 来降低时间复杂度。

cpp 复制代码

class Solution {
public:
    vector<int> dailyTemperatures(vector<int>& temperatures) {
        int n = temperatures.size();
        vector<int>ans(n), next(101, INT_MAX);
        for(int i = n - 1; i >= 0; i --){
            int index = INT_MAX;
            for(int t = temperatures[i] + 1; t <= 100; t ++){
                index = min(index, next[t]);
            }
            if(index != INT_MAX){
                ans[i] = index - i;
            }
            next[temperatures[i]] = i;
        }
        return ans;
    }
};

单调栈

栈中存储还没有被找到的满足条件的第 x 天。

如果是递增序列，栈中始终就一个元素，不断移入移出。

如果是非递增，则一直入栈，找到第一个大于栈顶的，再依次出栈判断。

cpp 复制代码

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