单调栈的用处
单调栈就是用来存放已经遍历过的元素。当解决问题需要记录前面的元素,这个时候就需要用到单调栈。比如求一个数组中一个元素后面第一个大于它的元素。可以使用暴力去遍历,但是会出现重复判断,性能不优。这个时候用一个单调栈来存放已经遍历过的元素。这样来看我们可知
单调栈的本质是空间换时间。根据题目要求,单调栈里面可以存放元素的下标也可以存放元素值。
题目
第一道是很经典的单调栈题目:
题目要求每一天温度后面第一个比该天温度大的天数。本质就是给定一个数组,求每一个元素后面第一个比该元素大的下标,最后返回的数组存放下标相减即可。若后面元素没有比它大的,那么直接赋值为0。
遍历给定的数组,有三种情况:(遍历这个的数组设为T,单调栈设为st)
1、T[i] > T[(st,top()] (st内存放的是数组下标)(大于的情况,显然满足题目要求。
这个天数应为: i - st.top()。
2、T[i] == T[(st,top()] (当后面两种情况,不满足题目要求,所有直接把 i 压栈)
3、T[i] < T[(st,top()]
>就弹出,<=就压栈。那么这样这个单调栈从栈顶到栈低就是递增的。
class Solution {
public:
vector<int> dailyTemperatures(vector<int>& temperatures) {
vector<int> res(temperatures.size(), 0);
//直接全部赋值为0,包括了没有大于这个元素的情况
stack<int> st;
st.push(0);
for(int i = 1; i < temperatures.size(); i++){
while(!st.empty() && temperatures[i] > temperatures[st.top()]){
//栈中不能为空
res[st.top()] = i - st.top();
st.pop();
}
st.push(i);
}
return res;
}
};