Leetcode739. 每日温度

Every day a Leetcode

题目来源：739. 每日温度

解法1：单调栈-从左到右

单调栈中记录还没算出「下一个更大元素」的那些数（的下标）。

代码：

/*
 * @lc app=leetcode.cn id=739 lang=cpp
 *
 * [739] 每日温度
 */

// @lc code=start

// 暴力
// Time Limit Exceeded

// class Solution
// {
// public:
//     vector<int> dailyTemperatures(vector<int> &temperatures)
//     {
//         int n = temperatures.size();
//         vector<int> answer(n, 0);
//         for (int i = 0; i < n - 1; i++)
//         {
//             int j = i + 1;
//             while (j < n && temperatures[i] >= temperatures[j])
//                 j++;
//             answer[i] = j == n ? 0 : j - i;
//         }
//         return answer;
//     }
// };

// 单调栈

class Solution
{
public:
    vector<int> dailyTemperatures(vector<int> &temperatures)
    {
        int n = temperatures.size();
        vector<int> answer(n, 0);
        stack<int> indices;
        for (int i = 0; i < n; i++)
        {
            while (!indices.empty())
            {
                int preIndex = indices.top();
                // 如果当前温度<=之前的温度，退出
                if (temperatures[i] <= temperatures[preIndex])
                    break;
                // 否则，之前温度对应下标的天数=当前下标i-之前下标preIndex
                indices.pop();
                answer[preIndex] = i - preIndex;
            }
            indices.push(i);
        }
        return answer;
    }
};
// @lc code=end

结果：

复杂度分析：

时间复杂度：O(n)，其中 n 为数组 temperatures 的长度。

空间复杂度：O(n)，其中 n 为数组 temperatures 的长度。注意这种写法栈中可以有重复元素。

解法2：单调栈-从右到左

单调栈中记录下一个更大元素的「候选项」。

代码：

// 单调栈-从右到左

class Solution
{
public:
    vector<int> dailyTemperatures(vector<int> &temperatures)
    {
        int n = temperatures.size();
        vector<int> ans(n);
        stack<int> st;
        for (int i = n - 1; i >= 0; i--)
        {
            int t = temperatures[i];
            // 当前温度大于等于之前的最大温度，小于等于当前温度的栈中温度全部全掉
            while (!st.empty() && t >= temperatures[st.top()])
                st.pop();
            if (!st.empty())
                ans[i] = st.top() - i;
            st.push(i);
        }
        return ans;
    }
};

结果：

复杂度分析：

时间复杂度：O(n)，其中 n 为数组 temperatures 的长度。

空间复杂度：O(min(n,U))，其中 U=max⁡(temperatures)−min⁡(temperatures)+1。返回值不计入，仅考虑栈的最大空间消耗。