每日温度
给定一个整数数组 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 <= 105
30 <= temperatures[i] <= 100
要用单调栈,思路也是比较灵活的一个(on²肯定不是允许的方法,所以肯定要降时间复杂度
思路就是从结尾倒着遍历,然后栈压入的是索引,且为单调递减栈(栈顶元素为当前栈内最小)
python
class Solution:
def dailyTemperatures(self, temperatures: List[int]) -> List[int]:
stack = []
n = len(temperatures)
ans = [0]*len(temperatures)
for i in range(n-1,-1,-1):
# 倒叙遍历
while stack and temperatures[stack[-1]]<=temperatures[i]:
stack.pop()
if stack:
ans[i] = stack[-1]-i
else: ans[i] = 0
stack.append(i)
return ans