给定一个整数数组 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]
思路:首先想到的肯定是暴力解法,两层for循环遍历,这样肯定能找到。但这样做当数据量很大的时候肯定会超时,时间复杂度是O(n^2)。
那怎么做才能降低时间复杂度呢,可以用一个叫做单调栈的东西。顾名思义,借助栈,并且这个栈是单调递增/单调递减的,具体操作如下:栈里放下标,然后遍历数组,当栈不为空,就一直与栈顶元素比大小,那么有3种情况
1.大于栈顶元素:那我们就找到了第一个比它大的元素,更新栈顶元素那个下标的值
2.小于栈顶元素:不操作,遍历数组下一个数
3.等于栈顶元素:不操作,因为得找大于它的数,遍历数组下一个数
这样做当找到第一个大的数的时候,会更新,当没有找到的时候不会操作,因此我们在初始化result的时候 就全部初始化为0.
代码(Python)
class Solution(object):
def dailyTemperatures(self, temperatures):
result = [0] * len(temperatures) #结果数组,初始全部置为0
stack = [0] #单调栈,初始放数组第一个数的下标,即0
for i in range(1,len(temperatures)):
while stack: #遍历单调栈里所有的元素
temp = stack.pop()
if temperatures[i] > temperatures[temp]: #如果找到了第一个大的数,更新result[temp]的值为i-temp
result[temp] = i - temp
else: #如果当前元素小于栈顶元素了,也不用去遍历栈里剩下的元素了,肯定都比当前元素大
stack.append(temp) #将弹出的元素添加回栈
break
stack.append(i) #将当前元素的下标放入栈
return result