请根据每日 气温 列表,重新生成一个列表。对应位置的输出为:要想观测到更高的气温,至少需要等待的天数。如果气温在这之后都不会升高,请在该位置用 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 (栈不为空 && 当前元素大于栈定元素)
{
// 循环判断更新
更新结果;
栈顶元素出栈;
}
// 找到了当前元素适当的插入位置(把比它小的都弹出去了)
当前数据入栈;
}
}