单调栈:满足单调性的栈结构
单调栈通常被用来解决求一个数左(右)边第一个小(大)于他的元素(元素的下标)。
暴力做法使用两层循环求解的时间复杂度是O(n2)。使用单调栈优化时间复杂度可以达到O(n)。
在指针i从左向右遍历的过程中,我们可以用一个栈来保存i左边的所有元素(不包括i指向元素),越大的元素越靠近栈顶,越小的元素越靠近栈底。所以我们每次从栈顶开始遍历,当栈不为空且栈顶元素大于a[i]时,tt --(弹出该元素),直到栈顶元素小于a[i]时,输出栈顶元素,并将a[i]压入栈顶。这样循环遍历数组我们就可以得到一个单调增栈,即栈顶元素最大,栈底元素最小。
由于每个元素一定会被压入栈且至多弹出一次,因此操作次数至多为2n,故时间复杂度为O(n)。
#include<iostream>
using namespace std;
const int N = 10010;
int stk[N],tt;
int n;
int main(){
cin >> n;
while(n -- ){
int x;
cin >> x;
while(tt && stk[tt] >= x) tt --; //当栈顶元素大于等于a[i]时,弹出栈顶元素
if(tt) cout << stk[tt] << ' '; //直到栈顶元素小于a[i]时,输出栈顶元素
else cout << -1 << ' ';
stk[ ++ tt] = x; //将a[i]压入栈顶
}
return 0;
}