题目:请根据每日 气温 列表,重新生成一个列表。对应位置的输出为:要想观测到更高的气温,至少需要等待的天数。如果气温在这之后都不会升高,请在该位置用 0 来代替。
例如,给定一个列表 temperatures = [73, 74, 75, 71, 69, 72, 76, 73],你的输出应该是 [1, 1, 4, 2, 1, 1, 0, 0]。
其实题意就是
:找到列表中,比当前项大的,最近的后一项。
两种解法:
一:
1,以当前项为起点,开始遍历之后的项。
2,遇到比当前大的,计算 index 的差,跳出循环
from typing import List
class Solution:
def dailyTemperatures(self, T: List[int]) -> List[int]:
size = len(T)
res = [0]*size
for i in range(size):
j = i+1
while j<size:
if T[j]<=T[i]:
j+=1
else:
res[i]=j-i
break
return res
二:
用栈来减少循环取值的次数
1,当栈为空,则从列表中取第一个元素,入栈
2,当栈不为空,则从栈中取出顶部元素a,从列表中取第一个元素b,比较。有两种情况:
* 栈顶元素a >= 列表元素b,此时,将a,b入栈
* 栈顶元素a < 列表元素b,此时,计算a与b的距离。3,继续从栈中取出元素,与b进行比较。直到不满足a<b。将b入栈。
直至结束。
from typing import List
class Solution:
def dailyTemperatures(self, T: List[int]) -> List[int]:
size = len(T)
stack = []
res = [0] * size
for i in range(size):
while len(stack) > 0:
j = stack.pop()
if T[j]<T[i]:
res[j]=i-j
else:
stack.append(j)
break
stack.append(i)
return res