739. 每日温度(单调栈)

请根据每日 气温 列表,重新生成一个列表。对应位置的输出为:要想观测到更高的气温,至少需要等待的天数。如果气温在这之后都不会升高,请在该位置用 0 来代替。

例如,给定一个列表 temperatures = [73, 74, 75, 71, 69, 72, 76, 73],你的输出应该是 [1, 1, 4, 2, 1, 1, 0, 0]。

提示:气温 列表长度的范围是 [1, 30000]。每个气温的值的均为华氏度,都是在 [30, 100] 范围内的整数。

分析:

想通过一次遍历数组来判断每个元素后面有没有比自己大的,而且还要记录和自己的下标相差多少,那么就需要有那么一个数据结构来维护下标信息,在这里用到了单调栈,单调栈就是像汉诺塔一样,要么满足下面的每个都比上面的大(单调递减栈),反之同理,在这个题中要的是后面的有没有比前面大的,理想情况是希望后面的都小,这样就避免对栈进行循环出栈的操作,所以构造的是递减栈,代码如下:

ps:之所以能这么写的还有一个原因是要找的是后面第一个比它大的

#include<stack>
class Solution {
    
    
public:
    vector<int> dailyTemperatures(vector<int>& T) {
    
    
        int len = T.size();
        if(len < 1) return vector<int>{
    
    };
        stack<int> st;
        vector<int> res(T.size(), 0);
        // 题目中要找的是每个气温后面有没有比它更高的,有就更新,没有就是0
        // 所以构造一个栈底大栈顶小的单调栈,这样满足res初始化值为0,只有后面的值比栈顶大才需更新res 
        for(int i = 0; i < len; i++){
    
    
            // 栈中存的是对应数组中元素值的下标
            if(st.empty() || T[i] <= T[st.top()]){
    
    
                st.push(i);
            }else{
    
    
                while(!st.empty() && T[i] > T[st.top()]){
    
    
                    res[st.top()] = i - st.top(); 
                    st.pop();
                }
                st.push(i);
            }
        }
        return res;
    }
};

ps:单调栈的构造为代码如下

stack<int> st;
vector<int> nums; // 一般伴随着要进行比较的数组
for (遍历这个数组)
{
    
    
	// 举例构造递减栈,先写不需要出栈操作的情况
	if (栈空 || 当前元素小于等于栈顶元素)
	{
    
    
		// 满足想法,直接如栈
		入栈;
	}
	else
	{
    
    
		// 栈不为空,避免溢出,当前元素大于栈定元素,不符合我们的初衷想法,需要出栈
		while (栈不为空 && 当前元素大于栈定元素)
		{
    
    
			// 循环判断更新
			更新结果;
			栈顶元素出栈;
			
		}
		// 找到了当前元素适当的插入位置(把比它小的都弹出去了)
		当前数据入栈;
	}
}

在这里插入图片描述

猜你喜欢

转载自blog.csdn.net/qq_34612223/article/details/114435036
今日推荐