LeetCode笔记 739. 每日温度

PiccoloYang

于 2020-10-21 23:04:26 发布

阅读量170

点赞数

CC 4.0 BY-SA版权

分类专栏： LeetCode笔记

本文链接：https://blog.csdn.net/qq_33183649/article/details/109211611

LeetCode笔记专栏收录该内容

15 篇文章

订阅专栏

本文介绍了一种巧妙利用栈的数据结构来解决温度变化问题的算法。通过对比当前温度与栈顶温度，找到需要等待的天数，展示了如何在不使用暴力遍历的条件下提高代码效率。代码实例和思路深入浅出，适合栈操作的编程练习。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

问题描述：

在这里插入图片描述

思路分析：

不考虑暴力遍历的方法，太笨比了，失去了刷题的意义。

这个题目是在栈相关的练习下，所以考虑借助栈来解决。
1、先从头开始遍历数组，依次将温度入栈；
2、如果当前温度大于栈顶温度，将栈顶温度出栈，且两个温度的下标之差就是所求的需要等待的天数。然后继续比较当前温度跟新的栈顶温度的大小，重复之前的操作，直到该温度小于栈顶元素，则将其入栈；
3、如果当前温度小于栈顶温度，则继续将下一个元素入栈，直到该温度大于栈顶温度，跳转至步骤2；
4、遍历直到数组结束；
其实上述步骤就是为了构建一个递减栈。

代码实现：

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