题目
给定一个整数数组 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 <= 10^5
30 <= temperatures[i] <= 100
解题方法
单调栈
单调栈通常是一维数组,用于寻找任一个元素的右边或者左边第一个比自己大或者小的元素的位置。
正向遍历,使用递增单调栈保存未处理的元素,在弹出时进行处理。
时间复杂度O(n)
,空间复杂度O(n)
C++代码:
反向遍历,使用递减单调栈保存处理过的元素,在入栈时处理。class Solution {
public:
vector<int> dailyTemperatures(vector<int>& temperatures) {
stack<int> next;
vector<int> result(temperatures.size(), 0);
for(int i=0; i<temperatures.size(); i++) {
while(!next.empty() && temperatures[i]>temperatures[next.top()]) {
result[next.top()] = i-next.top();
next.pop();
}
next.push(i);
}
return result;
}
};
C++代码:class Solution {
public:
vector<int> dailyTemperatures(vector<int>& temperatures) {
int size = temperatures.size();
stack<int> next;
vector<int> result(size, 0);
for(int i=size-1; i>=0; i--) {
while(!next.empty() && temperatures[i]>=temperatures[next.top()]) next.pop();
if(!next.empty()) {
result[i] = next.top()-i;
}
next.push(i);
}
return result;
}
};