什么是单调栈?
- 单调栈(Monotonic Stack)是一种数据结构,通常用于解决与单调性相关的问题。它是一个栈(Stack)的变种。
单调栈的本质是空间换时间,因为在遍历的过程中需要用一个栈来记录右边第一个比当前元素高的元素,优点是整个数组只需要遍历一次。
更直白来说,就是用一个栈来记录我们遍历过的元素,因为我们遍历数组的时候,我们不知道之前都遍历了哪些元素,以至于遍历一个元素找不到是不是之前遍历过一个更小的,所以我们需要用一个容器(这里用单调栈)来记录我们遍历过的元素。
怎么用单调栈
在使用单调栈的时候首先要明确如下几点:
- 单调栈里存放的元素是什么?
单调栈里只需要存放元素的下标i就可以了,如果需要使用对应的元素,直接T[i]就可以获取。
- 单调栈里元素是递增呢? 还是递减呢?
-
单调递增栈:在单调递增栈中,栈内元素从底部到顶部是递增的,即栈顶元素是最小的。这种栈通常用于找到下一个更大的元素。
-
单调递减栈:在单调递减栈中,栈内元素从底部到顶部是递减的,即栈顶元素是最大的。这种栈通常用于找到下一个更小的元素。
使用单调栈主要有三个判断条件
- 当前遍历的元素T[i]小于栈顶元素T[st.top()]的情况
- 当前遍历的元素T[i]等于栈顶元素T[st.top()]的情况
- 当前遍历的元素T[i]大于栈顶元素T[st.top()]的情况
给定一个整数数组 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]
class Solution {
public int[] dailyTemperatures(int[] t) {
// 创建答案数组
int[] res = new int[t.length];
Stack<Integer> st = new Stack<Integer>();
// 将0索引压入栈,因为0号元素循环开始前就有
st.push(0);
// 开始循环数组
for(int i = 1 ; i < t.length; i++){
// 内部开始三步判断
if(t[i]<t[st.peek()]) st.posh(i);
else if(t[i]==t[st.peek()]) st.push(i);
else{
// 注意判断循环终止条件
while(!st.isEmpty()&&t[i]>t[st.peek()]){
// 此时存的是下标
res[st.peek()] = i - st.peek();
st.pop();
}
// 当循环终止,需要继续压栈并进行比较
st.push(i);
}
}
return res;
}
}