1. 程式人生 > >739. Daily Temperatures(單調棧)

739. Daily Temperatures(單調棧)

題目:給出一個數組,對每個位置,在當前位置後面找出最近的比當前位置大的數,求出位置差。

思路:維護一個值單調遞減的棧(棧中的元素是陣列元素的下標),當前元素大於棧頂元素,將棧頂元素出棧,直到棧空或者當前元素小於棧頂元素為止。在結果陣列中記錄結果(當前下標減去棧頂元素)。更新當前元素。直到遍歷所有元素。

例如 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; } };