leetcode 棧739. 每日溫度
阿新 • • 發佈:2018-12-30
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; } };
感謝 :