Leetcode 739-Daily Temperatures（单调栈，与下一个最大/小元素之间的个数）

最新推荐文章于 2024-10-06 13:05:10 发布

祝如愿~

最新推荐文章于 2024-10-06 13:05:10 发布

阅读量45

点赞数

分类专栏： # 单调栈文章标签： leetcode 算法数据结构 c++

本文链接：https://blog.csdn.net/PriscillaZhao/article/details/128768179

版权

单调栈专栏收录该内容

6 篇文章 0 订阅

订阅专栏

Leetcode 739 Daily Temperatures

题干：

给定一个数组 $t e m p er a t u res$ ，存放的是近几天的天气气温，对于每一天，还要至少等多少天才能等到一个更暖和的气温，如果等不到就填 $0$ 。

Input: temperatures = [73,74,75,71,69,72,76,73]
Output: [1,1,4,2,1,1,0,0]

解：

单调栈模板题变形。

相当于对数组中的每个元素 $a [i]$ ，寻找它之后比它大的第一个元素的下标 $t$ ， $t - i$ 即为所求。

vector<int> dailyTemperatures(vector<int>& temperatures) {
        int n = temperatures.size();
        stack<int> st;
        vector<int> answer(n);
        for(int i = n-1; i >= 0; i--){
            while(!st.empty() && temperatures[st.top()] <= temperatures[i]){
                st.pop();
            }
            answer[i] = st.empty() ? 0: (st.top() - i);
            st.push(i);
        }
        return answer;
    }