题目链接
题目描述
给定一个整数数组 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 <= 10^5
30 <= temperatures[i] <= 100
解题思路
暴力法
- 使用双重循环找到符合题意的要求。复杂度为
O(n^2)
单调栈法
- 通常是一维数组,要寻找任一个元素的右边或者左边第一个比自己大或者小的元素的位置,此时我们就要想到可以使用单调栈了,时间复杂度为
O(n)
- 单调栈的本质是时间换空间,因为在遍历的过程中需要用一个栈来记录右边第一个比当前元素高的元素,优点是只需要遍历一次
- 在使用单调栈的时候首先需要明确以下几点:
- 单调栈里存放的元素是什么
- 单调栈里面只需要存放元素的下标
i
就可以了
- 单调栈里面只需要存放元素的下标
- 单调栈里的元素是递增还是递减
- 递增,因为只有递增的时候,加入一个元素
i
,才知道栈顶元素在数组中右面第一个比栈顶元素大的元素是i
- 递增,因为只有递增的时候,加入一个元素
- 使用单调栈主要右三个判断条件
- 当前遍历的元素
temperatures[i] < temperatures[stack.peek()]
- 入栈
- 当前遍历的元素
temperatures[i] = temperatures[stack.peek()]
- 入栈
- 当前遍历的元素
temperatures[i] > temperatures[stack.peek()]
- 将栈顶弹出,直到满足
temperatures[i] < temperatures[stack.peek()]
- 同时记录
ans[stack.peek()] = i - stack.peek()
- 将栈顶弹出,直到满足
- 当前遍历的元素
- 单调栈里存放的元素是什么
AC代码
暴力法
class Solution {
public int[] dailyTemperatures(int[] temperatures) {
int[] ans = new int[temperatures.length];
for (int i = 0; i < temperatures.length - 1; i++) {
for (int j = i + 1; j < temperatures.length; j++) {
if (temperatures[j] > temperatures[i]) {
ans[i] = j - i;
break;
}
}
}
return ans;
}
}
单调栈法
class Solution {
public int[] dailyTemperatures(int[] temperatures) {
int[] ans = new int[temperatures.length];
Stack<Integer> stack = new Stack<>();
for (int i = 0; i < temperatures.length; i++) {
while (!stack.isEmpty() && temperatures[i] > temperatures[stack.peek()]) {
ans[stack.peek()] = i - stack.peek();
stack.pop();
}
stack.push(i);
}
return ans;
}
}