739. 每日温度
题目描述
请根据每日 气温 列表,重新生成一个列表。对应位置的输出为:要想观测到更高的气温,至少需要等待的天数。如果气温在这之后都不会升高,请在该位置用 0 来代替。
例如,给定一个列表 temperatures = [73, 74, 75, 71, 69, 72, 76, 73],你的输出应该是 [1, 1, 4, 2, 1, 1, 0, 0]。
提示:气温 列表长度的范围是 [1, 30000]。每个气温的值的均为华氏度,都是在 [30, 100] 范围内的整数。
思路一:暴力双循环
从当前元素往后找,找到第一个大于自己的元素,保存下标之差
1 class Solution { 2 public int[] dailyTemperatures(int[] T) { 3 // 暴力双重循环 4 int len = T.length; 5 int[] arr = new int[len]; 6 for(int i = 0; i < len-1; i++){ 7 for(int j = i+1; j < len; j++){ // 跳过温度小于等于T[i]的 8 if(T[i] < T[j]){ 9 arr[i] = j - i; 10 break; 11 } 12 } 13 } 14 return arr; 15 } 16 }
leeetcode 执行用时:1248 ms > 5.01%, 内存消耗:46.7 MB > 45.47%
复杂度分析:
时间复杂度:有两个for循环,最坏时间复杂度为O(n2)
空间复杂度:除了用来存储结果的数组arr[], 其他的空间都是常量级别的,所以空间复杂度为O(1)
思路二:利用递减栈
遍历数组,判断当前元素与栈顶下标对应的元素的大小,如果如果当前元素大于栈顶下标对应的元素,弹出栈顶下标,计算当前元素和栈顶下标之差,保存在栈顶下标的结果数组中,否则将当前元素的下标也压入栈中,始终保持从栈底到栈顶下标对应的元素是递减的。
1 class Solution { 2 public int[] dailyTemperatures(int[] T) { 3 // 递减栈 4 int len = T.length; 5 int[] res = new int[len]; 6 Stack<Integer> stack = new Stack<Integer>(); 7 8 // 遍历数组 9 for(int i = 0; i < len; i++){ 10 // 如果栈不为空且栈顶下标对应的元素小于当前元素,持续弹栈,计算下标之差 11 while(!stack.isEmpty() && T[i] > T[stack.peek()]){ 12 int index = stack.pop(); 13 res[index] = i - index; 14 } 15 16 // 最后将当前元素的下标压入栈中 17 stack.push(i); 18 } 19 return res; 20 } 21 }
leetcode 执行用时:22 ms > 51.39%, 内存消耗:46.7 MB > 38.37%, 根据这个运行时间来看,确实是快了几十倍
复杂度分析:
时间复杂度:只遍历了一次数组,所以时间复杂度为O(n)
空间复杂度:需要一个栈,栈的最大大小可能为数组大小,所以空间复杂度为O(n)