根据每日 气温 列表,重新生成一个列表。对应位置的输出为:要想观测到更高的气温,至少需要等待的天数。如果气温在这之后都不会升高,请在该位置用 0 来代替。
例如,给定一个列表 temperatures = [73, 74, 75, 71, 69, 72, 76, 73],你的输出应该是 [1, 1, 4, 2, 1, 1, 0, 0]。
题目理解:数组中元素与下一个比它大的元素之间的距离
复杂度解析:该方法只需要对数组进行一次遍历,每个元素最多被压入和弹出堆栈一次,所以算法复杂度为O(N)。
维护递减栈,后入栈的元素总比栈顶元素小。
-
比对当前元素与栈顶元素的大小
-
若当前元素 < 栈顶元素:入栈
-
若当前元素 > 栈顶元素:弹出栈顶元素,记录两者下标差值即为所求天数
这里用栈记录的是 T 的下标。
Python:
class Solution(object):
def dailyTemperatures(self, T):
"""
:type T: List[int]
:rtype: List[int]
"""
stack = list()
t_length = len(T)
res_list = [0 for _ in range(t_length)]
for key, value in enumerate(T):
if stack:
while stack and T[stack[-1]] < value:
res_list[stack[-1]] = key - stack[-1]
stack.pop()
stack.append(key)
return res_list