思路
一开始想到的是暴力求解,每遍历一个温度,就往后寻找比它大的那个数,直到找到为止。这种方法不出意料,超时。
这题便签是栈,故用栈来解决。这里直接复制LeetCode的官方解法说明,与我的大同小异。
可以维护一个存储下标的单调栈,从栈底到栈顶的下标对应的温度列表中的温度依次递减。如果一个下标在单调栈里,则表示尚未找到下一次温度更高的下标。
正向遍历温度列表。对于温度列表中的每个元素 T[i],如果栈为空,则直接将 i 进栈,如果栈不为空,则比较栈顶元素 prevIndex 对应的温度 T[prevIndex] 和当前温度 T[i],如果 T[i] > T[prevIndex],则将 prevIndex 移除,并将 prevIndex 对应的等待天数赋为 i - prevIndex,重复上述操作直到栈为空或者栈顶元素对应的温度小于等于当前温度,然后将 i 进栈。
为什么可以在弹栈的时候更新 ans[prevIndex] 呢?因为在这种情况下,即将进栈的 i 对应的 T[i] 一定是 T[prevIndex] 右边第一个比它大的元素,试想如果 prevIndex 和 i 有比它大的元素,假设下标为 j,那么 prevIndex 一定会在下标 j 的那一轮被弹掉。
由于单调栈满足从栈底到栈顶元素对应的温度递减,因此每次有元素进栈时,会将温度更低的元素全部移除,并更新出栈元素对应的等待天数,这样可以确保等待天数一定是最小的。
以下用一个具体的例子帮助读者理解单调栈。对于温度列表 [73,74,75,71,69,72,76,73],单调栈 stack 的初始状态为空,答案ans 的初始状态是 [0,0,0,0,0,0,0,0],按照以下步骤更新单调栈和答案,其中单调栈内的元素都是下标,括号内的数字表示下标在温度列表中对应的温度。
- 当 i=0 时,单调栈为空,因此将 0 进栈。
- stack=[0(73)]
- ans=[0,0,0,0,0,0,0,0]
- 当 i=1时,由于 74 大于 73,因此移除栈顶元素 0,赋值 ans[0]:=1−0,将 1 进栈。
- stack=[1(74)]
- ans=[1,0,0,0,0,0,0,0]
- 当 i=2时,由于 75 大于 74,因此移除栈顶元素 1,赋值 ans[1]:=2−1,将 2 进栈。
- stack=[2(75)]
- ans=[1,1,0,0,0,0,0,0]
- 当 i=3 时,由于 71 小于 75,因此将 3 进栈。
- stack=[2(75),3(71)]
- ans=[1,1,0,0,0,0,0,0]
- 当 i=4 时,由于 69 小于 71,因此将 4 进栈。
- stack=[2(75),3(71),4(69)]
- ans=[1,1,0,0,0,0,0,0]
- 当 i=5 时,由于 72 大于 69 和 71,因此依次移除栈顶元素 4 和 3,赋值 ans[4]:=5−4 和 ans[3]:=5−3,将 5 进栈。
- stack=[2(75),5(72)]
- ans=[1,1,0,2,1,0,0,0]
- 当 i=6 时,由于 76 大于 72 和 75,因此依次移除栈顶元素 5 和 2,赋值 ans[5]:=6−5 和 ans[2]:=6−2,将 6 进栈。
- stack=[6(76)]
- ans=[1,1,4,2,1,1,0,0]
- 当 i=7 时,由于 73 小于 76,因此将 7 进栈。
- stack=[6(76),7(73)]
- ans=[1,1,4,2,1,1,0,0]
# python3
class Solution:
def dailyTemperatures(self, T: List[int]) -> List[int]:
l = len(T)
stack = list()
ans = [0] * l
for i in range(l):
for _ in range(len(stack)):
if T[i] > T[stack[-1]]:
ans[stack[-1]] = i - stack[-1]
stack.pop()
else:
stack.append(i)
break
stack.append(i)
return ans
执行用时:
644 ms, 在所有 Python3 提交中击败了12.90%的用户
内存消耗:
19.7 MB, 在所有 Python3 提交中击败了10.51%的用户
效率一般,做改进,将for循环简化成while:
# python3
class Solution:
def dailyTemperatures(self, T: List[int]) -> List[int]:
l = len(T)
stack = list()
ans = [0] * l
for i in range(l):
while stack and T[i] > T[stack[-1]]:
ans[stack[-1]] = i - stack[-1]
stack.pop()
stack.append(i)
return ans
执行用时:
480 ms, 在所有 Python3 提交中击败了99.00%的用户
内存消耗:
19 MB, 在所有 Python3 提交中击败了58.52%的用户