LeetCode#739. 每日温度 (python解法 单调栈)

问题描述:

根据每日 气温 列表,请重新生成一个列表,对应位置的输入是你需要再等待多久温度才会升高超过该日的天数。如果之后都不会升高,请在该位置用 0 来代替。

例如,给定一个列表 temperatures = [73, 74, 75, 71, 69, 72, 76, 73],你的输出应该是 [1, 1, 4, 2, 1, 1, 0, 0]。

提示:气温 列表长度的范围是 [1, 30000]。每个气温的值的均为华氏度,都是在 [30, 100] 范围内的整数。

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/daily-temperatures

 思路一:

一开始考虑使用递归的方法来实现。

写一个计算每个温度值对应的天数,如果小于等于当前温度,就递归,并且将临时的天数计数器加一,直到某天的温度比当前温度高。结束递归。

并且可以注意到最后一天的温度对应的天数一定是0。因此可以便于处理边界问题。

但由于这个题目的数据范围是:[1~30000]。所以当数据量较大之后,不出意外的超时了。代码记录在下面。

Python:

class Solution(object):
    def dailyTemperatures(self, T):
        """
        :type T: List[int]
        :rtype: List[int]
        """
        res = []
        l = len(T)
        def getday(i,j,count):
            if j==l:
                return 0
            if T[j]>T[i]:
                count+=1
                return count
            if T[j]<=T[i]:
                return getday(i,j+1,count+1)
        for i in range(l-1):
            j = i+1
            count = 0
            res.append(getday(i,j,count))
        res.append(0)
        return res

思路二:

这道题正确的解法应该是维护一个递减栈,后入栈的元素总比栈顶元素小。

  • 比对当前元素与栈顶元素的大小
    • 若当前元素 < 栈顶元素:入栈
    • 若当前元素 > 栈顶元素:弹出栈顶元素,记录两者下标差值即为所求天数

题目示例:

一开始是空栈:[ ]

元素73:压入。[(73,0)]其中0是73的下标,

元素74,弹出73,压入74。[(74,1)]也就是比73大的第一个元素是74,

元素75,弹出74,压入75。[(75,2)]也就是比74大的一个元素是75.

元素71,压入。[(75,2),(71,3)]

元素69,压入。[(75,2),(71,3), (69,4)]

元素72,弹出69,弹出71。即,比69大的下一个是72,比71大的下一个是72。[(75,2),(72,5)]

元素76,弹出72,75,说明比72,75大的均为76。[(76, 6)]

元素73,压入。[(76,6),(73,7)]

栈不是空的,所以剩余的两个元素是没有下一个更大元素,可以填入0。

class Solution(object):
    def dailyTemperatures(self, T):
        """
        :type T: List[int]
        :rtype: List[int]
        """
        stack = []
        res = [0]*len(T)
 
        for index, value in enumerate(T):
            # 空栈,直接把元素放入
            if len(stack) == 0:
                stack.append((index, value))
            else:
                # 扫描元素小于栈顶元素,入栈
                if value < stack[-1][1]:
                    stack.append((index, value))
                # 扫描元素大于栈顶元素,弹出栈顶元素并记录索引差值。
                # 之后再判断与新的栈顶元素大小关系
                else:
                    while (len(stack) > 0) and (value > stack[-1][1]):
                        index_pop, value_pop = stack.pop()
                        res[index_pop] = index - index_pop
                    stack.append((index, value))
 
        return res

猜你喜欢

转载自blog.csdn.net/paranior/article/details/104508120
今日推荐