1.题目描述
根据每日气温列表,请重新生成一个列表,对应位置的输入是你需要再等待多久温度才会升高超过该日的天数。如果之后都不会升高,请在该位置用 0 来代替。
例如,给定一个列表 temperatures = [73, 74, 75, 71, 69, 72, 76, 73],你的输出应该是 [1, 1, 4, 2, 1, 1, 0, 0]。
提示:气温 列表长度的范围是 [1, 30000]。每个气温的值的均为华氏度,都是在 [30, 100] 范围内的整数。
2. 解题思路
(法一)暴力解法
啊~~这种方法超过时间限制了
(法二)用栈
在遍历数组时用栈把数组中的数存起来,如果当前遍历的数比栈顶元素来的大,说明栈顶元素的下一个比它大的数就是当前元素。
3. 代码实现
法一:
class Solution:
def dailyTemperatures(T):
newArray = []
while(len(T) > 1):
flag = 0
element = T.pop(0)
for i in range(len(T)):
if(element < T[i]):
flag = 1
newArray.append(i+1)
break
if(flag == 0):
newArray.append(0)
newArray.append(0)
return newArray
法二:
class Solution(object):
def dailyTemperatures(self, T):
N = len(T)
stack = []
res = [0] * N
for i, t in enumerate(T):
while stack and stack[-1][0] < t:
oi = stack.pop()[1]
res[oi] = i - oi
stack.append((t, i))
return res