给定一个整数数组
temperatures
,表示每天的温度,返回一个数组answer
,其中answer[i]
是指对于第i
天,下一个更高温度出现在几天后。如果气温在这之后都不会升高,请在该位置用0
来代替。示例 1:
输入:temperatures
= [73,74,75,71,69,72,76,73] 输出: [1,1,4,2,1,1,0,0]示例 2:
输入: temperatures = [30,40,50,60] 输出: [1,1,1,0]示例 3:
输入: temperatures = [30,60,90] 输出: [1,1,0]提示:
1 <= temperatures.length <= 105
30 <= temperatures[i] <= 100
思路:
栈保存当前温度的下标,每次从数组中读取一个温度,然后将其与栈中的温度进行比较。
如果当前温度比栈顶温度高,循环遍历出栈,计算栈顶温度需要等待几天才会出现跟高的温度。如果栈中已经没有比当前温度低的温度,将此温度下标入栈。
public int[] dailyTemperatures(int[] temperatures) {
int[] answer = new int[temperatures.length];
Stack<Integer> stack = new Stack<>();
for (int i = 0; i < temperatures.length; i++) {
while (!stack.empty() && temperatures[i] > temperatures[stack.peek()]){
int pop = stack.pop();
answer[pop] = i - pop;
}
stack.push(i);
}
return answer;
}
时间复杂度:O(n)
空间复杂度:O(n)