739. 每日溫度

东岸發表於2024-05-09

給定一個整數陣列 temperatures ,表示每天的溫度,返回一個陣列 answer ,其中 answer[i] 是指對於第 i 天,下一個更高溫度出現在幾天後。如果氣溫在這之後都不會升高,請在該位置用 0 來代替。

示例 1:

輸入: temperatures = [73,74,75,71,69,72,76,73]
輸出: [1,1,4,2,1,1,0,0]
示例 2:

輸入: temperatures = [30,40,50,60]
輸出: [1,1,1,0]
示例 3:

輸入: temperatures = [30,60,90]
輸出: [1,1,0]

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

相關文章