1. 题目

请根据每日 气温 列表 temperatures ,重新生成一个列表,要求其对应位置的输出为:要想观测到更高的气温,至少需要等待的天数。如果气温在这之后都不会升高,请在该位置用 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]


提示:
1 <= temperatures.length <= 105
30 <= temperatures[i] <= 100

2. 题解

from typing import List


class Solution:
# 不行,显示超时,估计时间复杂度太高
def dailyTemperatures(self, temperatures: List[int]) -> List[int]:
result = []
for i in range(len(temperatures) - 1):
temp = 0
for j in range(i + 1, len(temperatures)):
temp += 1
if temperatures[j] > temperatures[i]:
break
elif temp == len(temperatures) - i - 1:
temp = 0
else:
continue
result.append(temp)
result.append(0)
return result

def dailyTemperatures_2(self, temperatures: List[int]) -> List[int]:
stack = []
count = len(temperatures)
ret = [0] * count
for i in range(count):
while stack and temperatures[i] > temperatures[stack[-1]]:
ret[stack[-1]] = i - stack[-1]
stack.pop()
stack.append(i)
return ret


if __name__ == "__main__":
s = Solution()
print(s.dailyTemperatures([73,74,75,71,69,72,76,73]))