一、题目
请根据每日 气温
列表 temperatures
,请计算在每一天需要等几天才会有更高的温度。如果气温在这之后都不会升高,请在该位置用 0
来代替。
点击查看原题
难度级别: 中等
二、思路
1)单调栈
使用暴力法求解思路是很明确的,双重循环,每遍历到一个元素,就向后找到第一次比它大的数字,找不到就为0
思考一下,可不可以用一种数据结构存储后面元素的递增序列,这样就可以直接找到后面元素出现刚好比当前元素的较大值,那就是单调栈,单调栈递减栈的特点:
- 如果当前元素比栈顶元素大,则出栈,直到栈顶元素比当前元素大
由于从后往前遍历,如果出现了更大的元素值,会截断后面比当前元素小的所有值
三、代码
1)单调栈
class Solution {
public int[] dailyTemperatures(int[] temperatures) {
Stack<Integer> stack = new Stack();
int[] ans = new int[temperatures.length];
for (int i = temperatures.length - 1; i >= 0; i--) {
while (!stack.isEmpty() && temperatures[stack.peek()] <= temperatures[i]) {
stack.pop();
}
ans[i] = stack.isEmpty() ? 0 : (stack.peek() - i);
stack.push(i);
}
return ans;
}
}
时间复杂度为O(n),空间复杂度为O(n)