LeetCode:每日温度【739】
题目描述
根据每日 气温
列表,请重新生成一个列表,对应位置的输入是你需要再等待多久温度才会升高的天数。如果之后都不会升高,请输入 0
来代替。
例如,给定一个列表 temperatures = [73, 74, 75, 71, 69, 72, 76, 73]
,你的输出应该是 [1, 1, 4, 2, 1, 1, 0, 0]
。
提示:气温
列表长度的范围是 [1, 30000]
。每个气温的值的都是 [30, 100]
范围内的整数。
题目分析
说实话,看到这个题的时候,我下意识的想法是,两个For循环就可以解决,因为是找到第一个大于当前温度的温度,所以并不认为时间复杂度为N的平方。并且代码也顺利通过了。可这毕竟是一道中等难度的题目,我们还是要进一步思考一下如何优化。
首先呢,我们不要从头到尾遍历,而是反过来,因为如果是从前向后遍历的话,先遍历第一天,我们需要记录下载第一天的温度,然后遍历到第二天,需要用第二天的温度和第一天左比较,发现第二天比第一天温度高,所以又要返回去修改第一天的结果,将其设置为1。
如果是从后向前遍历的话,先遍历第二天,我们需要记录下第二天的温度,然后遍历到第一天,需要用第一天和第二天的温度作比较,发现第一天的气温比第二天的低,此时我们可以直接修改第一天(也就是当前遍历的位置)的结果,将其设置为1。
所以从分析来看,从后向前遍历比较好一点,因为两者都是需要记录下一系列信息,但是从前向后遍历的话还需要返回去进行修改结果,但是从后向前遍历的话只需要修改当前的结果即可。
我们在栈中保存当前的气温的下标,[为什么不保存值呢?知道下标,我们用temperatures[i]就可以知道值啊,并且我们还多保存了一个位置信息]。栈顶温度一定是离当前温度最近的温度位置。
如果当前温度,大于栈顶温度的话,说明栈顶温度的位置不是下一个回升的温度,那我们就要更换栈顶,即出栈操作,一直比较,知道找到小于栈顶温度。接着当前温度小于栈顶温度,那输出应该是stack.peek()-i;
Java题解
public int[] dailyTemperatures(int[] temperatures) { int[] res = new int[temperatures.length]; Stack<Integer> stack = new Stack<>(); for (int i =temperatures.length-1;i>=0; i--){ while (!stack.isEmpty()&&temperatures[i]>=temperatures[stack.peek()]) stack.pop(); if(stack.isEmpty()) res[i] = 0; else res[i] = stack.peek()-i; stack.push(i); } return res; }
原文地址:https://www.cnblogs.com/MrSaver/p/9945619.html