leetcode739. 每日溫度/單調棧

題目

根據每日 氣溫 列表，請重新生成一個列表，對應位置的輸出是需要再等待多久溫度纔會升高超過該日的天數。如果之後都不會升高，請在該位置用 0 來代替。

例如，給定一個列表 temperatures = [73, 74, 75, 71, 69, 72, 76, 73]，你的輸出應該是 [1, 1, 4, 2, 1, 1, 0, 0]。

提示：氣溫 列表長度的範圍是 [1, 30000]。每個氣溫的值的均爲華氏度，都是在 [30, 100] 範圍內的整數。

來源：力扣（LeetCode）
鏈接：https://leetcode-cn.com/problems/daily-temperatures
著作權歸領釦網絡所有。商業轉載請聯繫官方授權，非商業轉載請註明出處。

基本思想1 : 單調棧

本題的本質是求與該元素後，第一個大於該元素的距離。
維護一個單調第減棧，當出現一個破壞該棧的元素時，說明該元素是第一個大於棧內的元素。

class Solution {
public:
    vector<int> dailyTemperatures(vector<int>& T) {
        if(T.size() == 0)
            return {};
        //單調棧,保證棧內的元素是單調遞減的
        stack<int> st;
        vector<int> res(T.size(), 0);
        st.push(0);
        for(int i = 1; i < T.size(); ++i){
            if(st.empty() || T[i] <= T[st.top()]){
                st.push(i);
            }
            else{
                while(!st.empty() && T[i] > T[st.top()]){
                    int pos = st.top();
                    st.pop();
                    res[pos] = i - pos;
                }
                st.push(i);
            }
        }
        return res;
    }
};

時間複雜度：O（n）,因爲每個元素最多進棧1次

基本思想2：暴力

也不算是完全暴力吧。

• 從後往前遍歷數組，維護一個next數組，數組的下標代表溫度，用來保存已經遍歷過的數組的位置
• 遍歷到該元素時，從比該元素大1的位置開始遍歷next數組，尋找最小下標

class Solution {
public:
    vector<int> dailyTemperatures(vector<int>& T) {
        if(T.size() == 0)
            return {};
        vector<int> next(101, INT_MAX);
        vector<int> res(T.size(), 0);
        for(int i = T.size() - 1; i >= 0; --i){
            int pos = INT_MAX;
            for(int j = T[i] + 1; j <= 100; ++j){
                pos = min(pos, next[j]);
            }
            if(pos != INT_MAX)
                res[i] = pos - i;
            next[T[i]] = i;
        }
        return res;
    }
};