LeetCode 739每日温度

LeetCode 739每日温度Medium

  • 题目简述:请根据每日 气温列表,重新生成一个列表。对应位置的输出为:要想观测到更高的气温,至少需要等待的天数。如果气温在这之后都不会升高,请在该位置用 0 来代替。

  • 输入:temperatures = [73, 74, 75, 71, 69, 72, 76, 73] 输出:[1, 1, 4, 2, 1, 1, 0, 0]

  • 思路:利用递减单调栈找到右边大于栈顶的元素,弹出栈顶直到满足单调递减,同时记录弹出元素与当前i的距离,保存在res数组中即是答案。每个元素仅出入栈一次,时间复杂度O(n)。

    • 依次遍历整个数组,先在空栈中压入第一个元素下标,然后用后面各个元素与当前栈顶元素比较,
      • 如果该元素比栈顶元素小,那么就将该元素下标加入栈中;
      • 如果该元素比栈顶元素大,那么就将栈顶元素弹出,然后用下一个栈顶元素比较,直到该元素小于栈顶或者栈为空,就将该元素加入栈中。保证了加入栈中的元素从栈底到栈顶依次递减
    • 每次栈中有元素弹出时,用数组记录每个弹出的栈顶元素下标对应的值,就是答案
class Solution {
public:
    vector<int> dailyTemperatures(vector<int>& T) {
        int n = T.size();
        vector<int> res(n);
        stack<int> st;
        for(int i = 0; i < n; i++)
        {
            while(!st.empty() && T[i] > T[st.top()])
            {
                res[st.top()] = i - st.top();
                st.pop();
            }
            st.push(i);
        }
        return res;
    }
};

图解参考:LeetCode 图解 | 739.每日温度

你可能感兴趣的:(LeetCode)