每日温度
请根据每日 气温 列表,重新生成一个列表。对应位置的输出为:要想观测到更高的气温,至少需要等待的天数。如果气温在这之后都不会升高,请在该位置用 0 来代替。
例如,给定一个列表 temperatures = [73, 74, 75, 71, 69, 72, 76, 73],你的输出应该是 [1, 1, 4, 2, 1, 1, 0, 0]。
提示:气温 列表长度的范围是 [1, 30000]。每个气温的值的均为华氏度,都是在 [30, 100] 范围内的整数。
思路:
- 我们使用1个最小栈来记录需要寻找最高气温的索引。
- 若栈为空,则将当前索引入栈。
- 若栈不为空,且栈顶索引对应的气温小于当前气温T[i],则说明栈顶的索引对应的气温找到了后续的最高气温。栈顶索引等待天数=i-s.top(),存在结果中,最后将栈顶索引pop出。
- 最后能找到后续高温的,都存在了res中,找不到的还存留在栈s中。
代码:
class Solution {
public:
vector<int> dailyTemperatures(vector<int>& T) {
vector<int> res(T.size(),0);
stack<int> s; //存放索引
for(int i=0;i<T.size();i++)
{
while(!s.empty() && T[s.top()]<T[i])
{
res[s.top()]=i-s.top();
s.pop();
}
s.push(i);
}
return res;
}
};
结果:
参考链接:
[1] 力扣官方题解:每日温度
[2] zpznba:LeetCode 739 每日温度
[3] zpznba:LeetCode 496 下一个更大元素