1. 问题描述
给定一个整数数组 temperatures ,表示每天的温度,返回一个数组 answer ,其中 answer[i] 是指在第 i 天之后,才会有更高的温度。如果气温在这之后都不会升高,请在该位置用 0 来代替。
示例:
输入: temperatures = [73,74,75,71,69,72,76,73]
输出: [1,1,4,2,1,1,0,0]输入: temperatures = [30,40,50,60]
输出: [1,1,1,0]输入: temperatures = [30,60,90]
输出: [1,1,0]
2. 解题思路
一开始没读懂题意,现在梳理一下:
以输入: temperatures = [73,74,75,71,69,72,76,73]
输出: [1,1,4,2,1,1,0,0]
为例:
answer[0] = 1 表示 0 天之后才有更高温度,指的是temperatures[1] = 74
answer[1] = 1 表示 1 天之后才有更高温度,指的是temperatures[2] = 75
answer[2] = 4 表示 4 天之后才有更高温度,指的是temperatures[6] = 76
answer[6] = 0 表示之后温度不再升高
answer[7] = 0 表示之后温度不再升高(最后一天总是0)
思路① 双重循环暴力求解
思路② 单调栈(方向:栈头 ----> 栈底):
栈内顺序:大 ➡ 小
入栈元素与当前栈内栈首元素比较:
若当前元素 大于 栈首元素,则二者下标做差
若当前元素 小于或等于 栈首元素,则入栈
3. 代码实现
双重循环暴力求解:
class Solution {
public int[] dailyTemperatures(int[] temperatures) {
int[] answer = new int[temperatures.length];
for (int i = 0; i < temperatures.length - 1; i++) {
for (int j = i + 1; j < temperatures.length; j++) {
if (temperatures[i] < temperatures[j]) {
answer[i] = j - i;
break;
} else {
answer[i] = 0;
}
}
}
return answer;
}
}
单调栈求解:
class Solution {
public int[] dailyTemperatures(int[] temperatures) {
Deque<Integer> stack = new LinkedList<>();
int[] answer = new int[temperatures.length];
for (int i = 0; i < temperatures.length; i++) {
// 栈不为空 且 入栈元素大于当前栈顶元素
while (!stack.isEmpty() && temperatures[i] > temperatures[stack.peek()]) {
int preIndex = stack.pop();
answer[preIndex] = i - preIndex;
}
// 放入的是元素下标
stack.push(i);
}
return answer;
}
}