1. 程式人生 > >leetcode 棧739. 每日溫度

leetcode 棧739. 每日溫度

739. 每日溫度

根據每日 氣溫 列表,請重新生成一個列表,對應位置的輸入是你需要再等待多久溫度才會升高的天數。如果之後都不會升高,請輸入 0 來代替。

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

提示:氣溫 列表長度的範圍是 [1, 30000]。每個氣溫的值的都是 [30, 100] 範圍內的整數。

直接按邏輯寫:

class Solution {
public:
    vector<int> dailyTemperatures(vector<int>& temperatures) {
        
        vector<int>my_vector;
        for(int i=0;i<temperatures.size();i++)
        {
            #注意這裡是<=
            for(int s=1;s<=temperatures.size();s++)
            {
               if((i+s)<temperatures.size()){
                   
                   
                     if(temperatures[i]<temperatures[i+s])
                    {
                
                        my_vector.push_back(s);
                        break;
                    }
               }
                else{
                    
                    my_vector.push_back(0);
                    break;   
                }
               
            }         
        } 
        return my_vector;
    }
};

35 / 37 個通過測試用例
狀態:超出時間限制

正解:

題意:輸入一個數組,代表每天的溫度,求解每天需要經過幾天會升溫,即需找陣列每個元素右邊第一個比自己大的數。利用單調棧性質建立單調遞減棧,遍歷陣列入棧,當將要入棧元素小於棧頂元素時入棧,若要入棧元素大於棧頂元素時,表示入棧元素為所求元素,記錄索引,彈出棧頂元素,將此元素壓棧,重新比較,一次迴圈時間複雜度O(n)。

樣例解析:

73入棧,74>73,73出棧,74入棧,74使得73出棧,所以73需要等待1-0=1天 ,棧內元素74.

75入棧,75>74, 74出棧,75入棧,75使得74出棧,所以74需要等待2-1=1天,棧內元素75.

71入棧,71<75,直接入棧,棧內元素 75,71

69直接入棧,棧內元素75,71,69

72入棧,72>69,69出棧,72使69出棧,所以69需要等待5-4=1天,此時棧內元素75,71,由與72>71,任不滿足單調棧性質,71出棧,72使71出棧,所以71需要等待5-3=2天,站內元素75,75>72,72直接入棧,棧內元素為:75,72.

以此類推

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



感謝 :