请根据每日 气温 列表,重新生成一个列表。对应位置的输出为:要想观测到更高的气温,至少需要等待的天数。如果气温在这之后都不会升高,请在该位置用 0 来代替。
例如,给定一个列表 temperatures = [73, 74, 75, 71, 69, 72, 76, 73],你的输出应该是 [1, 1, 4, 2, 1, 1, 0, 0]。
提示:气温 列表长度的范围是 [1, 30000]。每个气温的值的均为华氏度,都是在 [30, 100] 范围内的整数。
思路:这道题拿到手里知道是用栈,但是第一反应是存到栈里面的是温度值,但是这道题要求的只要是数组元素距离间的差值,也就是下标的差值,所以正确思路应该是入栈出栈的是数组元素的下标。
令int i=0,进入循环:
当栈为空或T[i]小于当前栈顶元素的温度值 T[栈顶]时,i入栈。i++。
当T[i]大于当前栈顶元素的温度值(T[栈顶]) 时,说明当前i对应的温度T[i] 比栈顶元素大,于是就pop,并用当前元素下标 i 减去 pop出的栈顶,得到下标的差 i-栈顶。即为从这天开始想要观测到更高的气温所需等待的天数。
还有一种暴力法,这种暴力解法就不写了,真正面试的时候估计也不会让用暴力法解。
class Solution {
public int[] dailyTemperatures(int[] T) {
Stack<Integer> stack = new Stack<>();
int n = T.length;
int[] days = new int[n];
int i = 0;
while(i<n)
{
if(stack.empty()||(T[stack.peek().intValue()]>=T[i]))
{
stack.push(i);
i++;
continue;
}
if(T[stack.peek().intValue()]<T[i])
{
int pop = stack.pop();
days[pop] = i-pop;
}
}
return days;
}
}