739. Daily Temperatures(單調棧)
阿新 • • 發佈:2019-01-23
題目:給出一個數組,對每個位置,在當前位置後面找出最近的比當前位置大的數,求出位置差。
思路:維護一個值單調遞減的棧(棧中的元素是陣列元素的下標),當前元素大於棧頂元素,將棧頂元素出棧,直到棧空或者當前元素小於棧頂元素為止。在結果陣列中記錄結果(當前下標減去棧頂元素)。更新當前元素。直到遍歷所有元素。
例如 temperatures = [73, 74, 75, 71, 69, 72, 76, 73] .
下標 棧 結果陣列 0 0 0 0 0 0 0 0 0 0 1 1 1 0 0 0 0 0 0 0 2 2 1 1 0 0 0 0 0 0 3 2 3 1 1 0 0 0 0 0 0 4 2 3 4 1 1 0 0 0 0 0 0 5 2 5 1 1 0 2 1 0 0 0 6 6 1 1 4 2 1 1 0 0 7 6 7 1 1 4 2 1 1 0 0
程式碼
class Solution {
public:
vector<int> dailyTemperatures(vector<int>& temperatures) {
stack<int>s;
vector<int>v(temperatures.size(),0);
for(int x=0;x<temperatures.size();x++){
while(!s.empty()&&temperatures[s.top()]<temperatures[x]){
v[s.top()] = x - s.top();
s.pop();
}
s.push(x);
}
while (!s.empty()){
v[s.top()] = 0;
s.pop();
}
return v;
}
};