739. 每日温度

文章目录

• • 题目
  • 代码
  • 原理图
  • 原理解释

题目

栈：739. 每日温度

给定一个整数数组 temperatures ，表示每天的温度，返回一个数组 answer ，其中 answer[i] 是指对于第 i 天，下一个更高温度出现在几天后。如果气温在这之后都不会升高，请在该位置用 0 来代替。

示例 1:

输入: temperatures = [73,74,75,71,69,72,76,73]

输出: [1,1,4,2,1,1,0,0]

示例 2:

输入: temperatures = [30,40,50,60]

输出: [1,1,1,0]

示例 3:

输入: temperatures = [30,60,90]

输出: [1,1,0]

提示：

1 <= temperatures.length <= 105

30 <= temperatures[i] <= 100

代码

class Solution {
public:
    vector<int> dailyTemperatures(vector<int>& temperatures) {
        stack<int>st;//栈用来存放数组下标
        vector<int>res(temperatures.size(),0);//存放结果
        for(int i=0;i<temperatures.size();i++){//遍历一遍数组
            while(!st.empty()&&temperatures[i]>temperatures[st.top()]){//若栈不为空并且当前元素比栈顶元素大则当前元素的下标i减去栈顶下标则为栈顶下标对应的下次温度比自己高的天数
                int index=st.top();//取栈顶
                res[index]=i-index;
                st.pop();
            }
            st.push(i);//若栈为空或不比栈顶大则压入栈
        }
        return res;
    }
};

原理图

方法1：

方法2：

原理解释

提示：算法流程及解释在代码中已标注

方法1：使用两个循环

对于第一个数，使用第二个循环依次遍历该元素后面的数，找到第一个比他它大的数，计数然后保存到当前数组位置。后面依次执行，时间复杂度为O(n方)

方法2：对于方法一的优化，一般使用空间换时间的方法

1、空间：使用单调栈 时间复杂度：O(n)

2、维护一个只存数组下标的单调递减栈

3、栈里的下标对应的温度，从栈底到栈顶一直递减

4、遍历温度数组：

当前温度 > 栈顶下标对应的温度 进入while循环下标与栈顶相见为天数

当前温度 ≤ 栈顶下标对应的温度 ，不会进入while循环，把当前下标压入栈，等待后面更大的温度。