管科研一新生首刷leetcode,本题思路来源leetcode官解,写此帖初衷是为了帮助自己更好的理解和“背诵”代码,方便后期复习,希望各位大佬多多指教!
题目描述
给定一个整数数组 temperatures ,表示每天的温度,返回一个数组 answer ,其中 answer[i] 是指对于第 i 天,下一个更高温度出现在几天后。如果气温在这之后都不会升高,请在该位置用 0 来代替。
输入: temperatures = [73,74,75,71,69,72,76,73]
输出: [1,1,4,2,1,1,0,0]
最近在学习栈与深度优先搜索,题目出现在栈知识点的推荐题目中,虽然知道本题要使用栈来求解,但是想了半天还是不知道怎么用栈实现,故打算先使用暴力求解,但是最简单粗暴的方式出现了超出时间限制的情况。
class Solution:
def dailyTemperatures(self, temperatures: List[int]) -> List[int]:
n = len(temperatures)
answers = [0] * n
for i in range(n-1):
for j in range(i+1,n):
if temperatures[i]<temperatures[j]:
answers[i] = j-i
break
return answers
能力所限,只能查看leetcode的题解。
解题思路
创建一个与temperatures等长的全零数组answer用来存放答案,创建一个空栈辅助本题操作。对temperatures数组进行遍历,当栈不为空,且当前遍历元素大于栈顶元素,则取出栈顶元素(temp),answer数组temp索引位置的值赋为当前遍历元素所在索引与栈顶元素(temp)的差值;否则,将当前遍历元素的索引加入栈,直到上述循环结束。
代码实现
class Solution:
def dailyTemperatures(self, temperatures: List[int]) -> List[int]:
n = len(temperatures)
answer = [0] * n
stack = []
for i in range(n):
while stack and temperatures[i] > temperatures[stack[-1]]:
temp = stack.pop()
answer[temp] = i - temp
stack.append(i)
return answer
当时看了答案过了好久复习才打算梳理一下思路,忘记参考的到底是哪篇文章了,貌似是这个。https://leetcode.cn/problems/daily-temperatures/solution/739mei-ri-wen-du-python3-by-leetcode-wei-dzpu/