题目描述
题目来源于leetcode:https://leetcode-cn.com/explore/learn/card/queue-stack/218/stack-last-in-first-out-data-structure/879/
每日温度
请根据每日 气温 列表,重新生成一个列表。对应位置的输出为:要想观测到更高的气温,至少需要等待的天数。如果气温在这之后都不会升高,请在该位置用 0 来代替。
例如,给定一个列表 temperatures = [73, 74, 75, 71, 69, 72, 76, 73],你的输出应该是 [1, 1, 4, 2, 1, 1, 0, 0]。
提示:气温 列表长度的范围是 [1, 30000]。每个气温的值的均为华氏度,都是在 [30, 100] 范围内的整数。
分析
此题有两种解法
第一种解法思路如下:
我们对每一天的温度查找这天之后的温度有没有大于它的,如果有,就将结果数组对应的值置为他们间隔的天数,如果没有将结果数组置0。这其中要用到双循环,第一个循环用来遍历所有天数,第二个循环用来寻找这天之后的温度。
第二种解法需要用到栈:
我们定义一个栈用来存储数组的下标,首先把0入栈,然后进行比较T[1]与
数组下标是栈中的顶部对应的天数的温度,也就是T[0],如果T[1]>T[0],就将结果数组下标为栈中的顶部的值对应的值置为1-0,并且出栈;其他结果就将下标入栈。
代码
第一种
public class Solution {
public int[] dailyTemperatures(int[] T) {
int[] result = new int[T.length];
/*
*使用双循环,第一个循环用来遍历所有天数,第二个循环用来
*查找这天之后的温度。
*/
for(int i=0;i<T.length;i++) {
int flag = 0;//定义一个标志
for(int j=i+1;j<T.length;j++) {
if(T[j]>T[i]) {
result[i] = j-i;
flag = 1;//如果这天以后有大于它的温度的天数,将标志位置为1
break;
}
}
if(flag == 0) {//标志位为0说明这天之后的温度都比这天低
result[i] = 0;
}
}
return result;
}
}
第二种
class Solution {
public int[] dailyTemperatures(int[] T) {
int[] result = new int[T.length];
Stack<Integer> stack = new Stack<>();
for(int i=0;i<T.length;i++) {
while(!stack.empty() && T[i]>T[stack.peek()]) {
int j = stack.pop();
result[j] = i-j;
}
stack.push(i);
}
return result;
}
}