leetcode-739.每日温度-单调栈

739. 每日温度

739.每日温度 - 力扣
解题思路:单调栈

题目:

给定一个整数数组 temperatures ,表示每天的温度,返回一个数组 answer ,其中 answer[i] 是指在第 i 天之后,才会有更高的温度。如果气温在这之后都不会升高,请在该位置用 0 来代替。

示例:

输入: temperatures = [73,74,75,71,69,72,76,73]
输出: [1,1,4,2,1,1,0,0]

解析:

​ 这是接触的第一道单调栈类型的题目。

​ 单调栈的主要思想是:栈中元素保持单调递增或者单调递减。为了保持这种单调性,需要(以单调递减为例)遇到一个大于栈顶元素的数时,不断pop栈中元素,直到栈顶元素大于新遇到的元素,然后将新元素push进栈;遇到一个小于栈顶元素的数时,就直接将新的元素push进栈。

​ 单调栈的经典题目是找到第i个元素之后第一个大于它的下标,此题与739不能说毫无相干,只能说一模一样。下面首先讲解单调栈的经典题目思路。

O(n^2)的做法是,遍历所有元素,再遍历此元素之后的所有元素,找到第一个大于它的元素下标。但这种方法时间复杂度过高,有没有O(n)的做法,只需遍历一次就完成题目呢?

​ 答案是:有!

​ 不过首先要明确对于遍历的作用。在O(n^2)的做法中,外层遍历是在确定寻找第一个大于当前元素的开始下标,然后再到内层循环中去真正地寻找。使用这种思路必定需要O(n^2)的时间复杂度。现在我们换一种思路,让循环的index表示以当前元素作为第一个大于的元素下标,然后去找在index之前的没有找到第一个大于它的元素的下标的、小于index对应元素的元素下标(这句话有点拗口)。那么我们就需要一个数据结构来存储【在index之前的没有找到第一个大于它的元素的下标的、小于index对应元素的元素下标】。

​ 单调栈这个数据结构刚好可以满足我们的需求。使用一个单调递减的单调栈,若新的元素(循环的index对应的元素)大于栈顶元素,那么对于栈顶元素来说,它就找到了目标的下标,所以记录它的NGE(Next Greater Element)后就将它pop出,然后判断新的栈顶元素是否也小于新的元素。最后还需要将新元素push进栈,因为新元素也需要找到它的NGE。若新的元素小于栈顶元素,那么说明没有元素以它为NGE,就不pop栈的任何元素,同时将新元素push进栈,因为新元素也需要找到它的NGE。

​ 至此,经典题的思路以及很清晰了,核心思想就在于使用单调栈来存储尚未找到NGE的元素下标

​ 而739.每日温度这道题,和经典题的唯一区别就在于,它找的不是第i个元素之后第一个大于它的下标,而是第一个大于它的元素与它的距离,那么只需要在记录时记录距离就好了。

代码:
使用javascript实现

var dailyTemperatures = function(temperatures) {
    var ans = Array(temperatures.length).fill(0);
    var stack = [];
    for(let i=0; i<temperatures.length; i++){
        while(stack.length!=0 && temperatures[stack[stack.length-1]]<temperatures[i]){
            ans[stack[stack.length-1]] = i-(stack[stack.length-1]);// 与经典题目唯一的区别。
            stack.pop();
        }
        stack.push(i);
    }
    return ans;
};
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值