快轉到主要內容
  1. Leetcode/

[LeetCode] 739. Daily Temperatures

·
LeetCode monotonic stack
目錄

題目連結:739. Daily Temperatures

解題思路
#

monotonic stack 的經典問題,尋找右方第一個比自己大的值,維護一個遞減的單調堆疊,pop的時候更新答案即可。

程式碼
#

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