LeetCode刷题之739. 每日温度
我不知道将去向何方,但我已在路上! |
---|
时光匆匆,虽未曾谋面,却相遇于斯,实在是莫大的缘分,感谢您的到访 ! |
- 题目:
根据每日气温列表,请重新生成一个列表,对应位置的输入是你需要再等待多久温度才会升高超过该日的天数。如果之后都不会升高,请在该位置用 0 来代替。例如,给定一个列表 temperatures = [73, 74, 75, 71, 69, 72, 76, 73],你的输出应该是 [1, 1, 4, 2, 1, 1, 0, 0]。 - 提示:
气温列表长度的范围是 [1, 30000]。每个气温的值的均为华氏度,都是在 [30, 100] 范围内的整数。 - 代码:
class Solution:
def dailyTemperatures(self, T: List[int]) -> List[int]:
stack = []
result = [0] * len(T)
for index in range(len(T)):
while stack != [] and T[stack[-1]] < T[index]:
result[stack[-1]] = index - stack[-1]
stack.pop()
stack.append(index)
return(result)
- 算法说明:
建立一个空栈,遍历T中的元素,如果当前元素小于栈顶元素,将当前元素的索引入栈;如果当前元素大于栈顶元素,弹出栈顶元素,求出当前索引和栈顶索引之间的差值,然后用差值更新列表result中栈顶元素对应位置的值。