739. 每日温度

给定一个整数数组 temperatures ,表示每天的温度,返回一个数组 answer ,其中 answer[i] 是指对于第 i 天,下一个更高温度出现在几天后。如果气温在这之后都不会升高,请在该位置用 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]

思路:首先想到的肯定是暴力解法,两层for循环遍历,这样肯定能找到。但这样做当数据量很大的时候肯定会超时,时间复杂度是O(n^2)。

那怎么做才能降低时间复杂度呢,可以用一个叫做单调栈的东西。顾名思义,借助栈,并且这个栈是单调递增/单调递减的,具体操作如下:栈里放下标,然后遍历数组,当栈不为空,就一直与栈顶元素比大小,那么有3种情况

1.大于栈顶元素:那我们就找到了第一个比它大的元素,更新栈顶元素那个下标的值

2.小于栈顶元素:不操作,遍历数组下一个数

3.等于栈顶元素:不操作,因为得找大于它的数,遍历数组下一个数

这样做当找到第一个大的数的时候,会更新,当没有找到的时候不会操作,因此我们在初始化result的时候 就全部初始化为0.

代码(Python)

class Solution(object):
    def dailyTemperatures(self, temperatures):
        result = [0] * len(temperatures)          #结果数组,初始全部置为0
        stack = [0]                               #单调栈,初始放数组第一个数的下标,即0
        for i in range(1,len(temperatures)):
            while stack:                          #遍历单调栈里所有的元素
                temp = stack.pop()
                if temperatures[i] > temperatures[temp]:  #如果找到了第一个大的数,更新result[temp]的值为i-temp
                    result[temp] = i - temp
                else:                             #如果当前元素小于栈顶元素了,也不用去遍历栈里剩下的元素了,肯定都比当前元素大
                    stack.append(temp)            #将弹出的元素添加回栈
                    break
            stack.append(i)                       #将当前元素的下标放入栈
        return result
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值