请根据每日 气温 列表 temperatures ,请计算在每一天需要等几天才会有更高的温度。如果气温在这之后都不会升高,请在该位置用 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
通过次数223,384提交次数327,789
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/daily-temperatures
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
题目分析
很明显的一道单调栈题目。评论区里还有源于KMP算法的更优解,我待会去看看。
这道题里的单调栈存储的是位置,为什么呢?很简单,因为题目要求的输出是位置。
AC代码
class Solution {
public:
vector<int> dailyTemperatures(vector<int>& temperatures) {
int N = temperatures.size();
stack<int> indices;
vector<int> res(N,0);
for(int i = 0;i<N;i++){
if(indices.empty()||temperatures[indices.top()]>temperatures[i]){
indices.push(i);
}
else{
while(!indices.empty()&&temperatures[indices.top()]<temperatures[i]){
res[indices.top()] = i - indices.top();//等待天数
indices.pop();
}
indices.push(i);
}
}
return res;
}
};