代码随想录单调栈——每日温度

题目

请根据每日 气温 列表,重新生成一个列表。对应位置的输出为:要想观测到更高的气温,至少需要等待的天数。如果气温在这之后都不会升高,请在该位置用 0 来代替。

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

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

思路

本题要找到右边第一个比自己大的元素

单调栈的使用场景:

通常是一维数组,要寻找任一个元素的右边或者左边第一个比自己大或者小的元素的位置,此时就要想到可以用单调栈了

单调栈要注意的几点:

  1. 单调栈存放的是元素的下标
  2. 明确单调栈的维护顺序(递增/递减)
  3. 判断当前遍历的元素和栈顶元素的关系
    ①当前元素 < 栈顶元素
    ②当前元素 == 栈顶元素
    ③当前元素 > 栈顶元素

在本题中,因为要找到右边第一个比自己大的元素

所以维护递减栈,即在栈中,后入栈的元素总比先入栈的元素小,本题中就是从栈底到栈顶(头)的下标对应的温度列表中的温度依次递减,这样的话可以保证,如果最后有下标在单调栈里,则表示尚未找到下一次温度更高的下标,因为如果有更高的温度,那么就会将栈顶元素出栈

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

算法思想:
正向遍历温度列表,对于温度列表中的每个元素 temperatures[i]
(1)如果栈为空,则直接将 i 进栈
(2)如果栈不为空,则比较栈顶元素 prevIndex 对应的温度temperatures[prevIndex] 和当前温度 temperatures[i];如果 temperatures[i] > temperatures[prevIndex](当前元素>栈顶元素,比较的是下标对应的元素),则将 prevIndex 移除(出栈的是下标),并将 prevIndex 对应的等待天数赋为 i - prevIndex(两者的下标差),重复上述操作直到栈为空或者栈顶元素对应的温度小于等于当前温度,然后将 i 进栈

java代码如下:


class Solution {
    public int[] dailyTemperatures(int[] temperatures) {
        int len = temperatures.length;
        int[] ans = new int[len];
        Deque<Integer> stack = new LinkedList<Integer>();//Stack<Integer> stack = new Stack<>()也可以
        for (int i = 0; i < len; i++) {
            int temperature = temperatures[i];
            while (!stack.isEmpty() && temperature > temperatures[stack.peek()]) {//因为单调栈存的是元素下标,所以temperatures[stack.peek()]表示的是当前stack.peek()的下标对应的温度
                int prevIndex = stack.pop();//将当前栈顶元素出栈
                ans[prevIndex] = i - prevIndex;//计算两个下标天数差,并存入结果
            }
            stack.push(i);
        }
        return ans;
    }
}

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

HDU-五七小卡

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值