问题描述:
根据每日 气温 列表,请重新生成一个列表,对应位置的输入是你需要再等待多久温度才会升高超过该日的天数。如果之后都不会升高,请在该位置用 0 来代替。
例如,给定一个列表 temperatures = [73, 74, 75, 71, 69, 72, 76, 73],你的输出应该是 [1, 1, 4, 2, 1, 1, 0, 0]。
提示:气温 列表长度的范围是 [1, 30000]。每个气温的值的均为华氏度,都是在 [30, 100] 范围内的整数。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/daily-temperatures
思路一:
一开始考虑使用递归的方法来实现。
写一个计算每个温度值对应的天数,如果小于等于当前温度,就递归,并且将临时的天数计数器加一,直到某天的温度比当前温度高。结束递归。
并且可以注意到最后一天的温度对应的天数一定是0。因此可以便于处理边界问题。
但由于这个题目的数据范围是:[1~30000]。所以当数据量较大之后,不出意外的超时了。代码记录在下面。
Python:
class Solution(object):
def dailyTemperatures(self, T):
"""
:type T: List[int]
:rtype: List[int]
"""
res = []
l = len(T)
def getday(i,j,count):
if j==l:
return 0
if T[j]>T[i]:
count+=1
return count
if T[j]<=T[i]:
return getday(i,j+1,count+1)
for i in range(l-1):
j = i+1
count = 0
res.append(getday(i,j,count))
res.append(0)
return res
思路二:
这道题正确的解法应该是维护一个递减栈,后入栈的元素总比栈顶元素小。
- 比对当前元素与栈顶元素的大小
- 若当前元素 < 栈顶元素:入栈
- 若当前元素 > 栈顶元素:弹出栈顶元素,记录两者下标差值即为所求天数
题目示例:
一开始是空栈:[ ]
![](/qrcode.jpg)
元素73:压入。[(73,0)]其中0是73的下标,
元素74,弹出73,压入74。[(74,1)]也就是比73大的第一个元素是74,
元素75,弹出74,压入75。[(75,2)]也就是比74大的一个元素是75.
元素71,压入。[(75,2),(71,3)]
元素69,压入。[(75,2),(71,3), (69,4)]
元素72,弹出69,弹出71。即,比69大的下一个是72,比71大的下一个是72。[(75,2),(72,5)]
元素76,弹出72,75,说明比72,75大的均为76。[(76, 6)]
元素73,压入。[(76,6),(73,7)]
栈不是空的,所以剩余的两个元素是没有下一个更大元素,可以填入0。
class Solution(object):
def dailyTemperatures(self, T):
"""
:type T: List[int]
:rtype: List[int]
"""
stack = []
res = [0]*len(T)
for index, value in enumerate(T):
# 空栈,直接把元素放入
if len(stack) == 0:
stack.append((index, value))
else:
# 扫描元素小于栈顶元素,入栈
if value < stack[-1][1]:
stack.append((index, value))
# 扫描元素大于栈顶元素,弹出栈顶元素并记录索引差值。
# 之后再判断与新的栈顶元素大小关系
else:
while (len(stack) > 0) and (value > stack[-1][1]):
index_pop, value_pop = stack.pop()
res[index_pop] = index - index_pop
stack.append((index, value))
return res