一、题目描述
给定一个整数数组 temperatures ,表示每天的温度,返回一个数组 answer ,其中 answer[i] 是指对于第 i 天,下一个更高温度出现在几天后。如果气温在这之后都不会升高,请在该位置用 0 来代替。
提示:
二、解题思路与代码实现
(1)暴力接法
将温度从后往前遍历,然后对每个温度值,遍历[该温度值,100]这个区间内的温度所出现的离当前温度最近的温度所对应下标值
//法2:暴力解法:小范围的数字可以用数组给对应位置赋值
/**
* 核心:
* (1)对所给温度从后往前遍历
* (2)next数组内的值为每个温度对应下标,且充分利用温度在30~100之间的特性
*
*/
public int[] dailyTemperatures01(int[] temperatures) {
int length = temperatures.length;
int[] ans = new int[length];
// next数组中保存当前温度对应的下标,便于后面找离当前温度最近的温度的下标
int[] next = new int[101];
// 给数组的每个位置填充相应值
Arrays.fill(next, Integer.MAX_VALUE);
// 从后往前遍历
for (int i = length - 1; i >= 0; --i) {
int warmerIndex = Integer.MAX_VALUE;
for (int t = temperatures[i] + 1; t <= 100; ++t) {
// 找到temperatures中,高于当前的temperatures[i],但下标又最接近temperatures[i]的温度下标
if (next[t] < warmerIndex) {
warmerIndex = next[t];
}
}
if (warmerIndex < Integer.MAX_VALUE) {
ans[i] = warmerIndex - i;
}
// 重点!
next[temperatures[i]] = i;
}
return ans;
}
(2)单调栈
栈内存储温度下标值。且栈内元素下标值所对应的温度从栈底到栈顶依次递减。如果当前入栈的元素比栈顶元素大,则出栈 ;元素能入栈表明当前元素必定比栈顶元素要小。
/**
* 法3:利用单调栈:栈内存储各个元素的下标,从栈底到栈顶的下标对应的元素大小依次递减,如果当前入栈的元素比栈顶元素大,则出栈
* 元素能入栈表明当前元素必定比栈顶元素要小
*/
public int[] dailyTemperatures02(int[] temperatures) {
int len = temperatures.length;
Deque<Integer> stack = new LinkedList<>();
int[] ans = new int[len];
for (int i = 0; i < len; i++) {
while (!stack.isEmpty() && temperatures[i] > temperatures[stack.peek()]) {
Integer preIndex = stack.pop();
ans[preIndex] = i - preIndex;
}
stack.push(i);
}
return ans;
}