目录
单调队列
题目描述
现在有一堆数字共N个数字(N<=10^6),以及一个大小为k
的窗口。现在这个从左边开始向右滑动,每次滑动一个单
位,求出每次滑动后窗口中的最大值和最小值。
显然,可以用线段树或者RMQ做。但是单调队列时间,空间复杂度更优。
基本操作:
单调队列一般用 deque 实现,存放结构体(因为需要值 和 下标)或 直接存放下标,但是比较的时候注意 索引回原数组。主要有两个操作:删头和去尾
1.删头
如果队列头的元素离开了我们当前操作的区间,那么这个元素就没有任何用了,直接 pop_front() 即可
2.去尾
假设即将进入队列的元素为 X ,队列尾指针为tail ,这时我们要比较二者的大小:
1. X < q[tail] 或者为 空
此时q仍然保证着递减性,故直接将 X 插入队列尾
那么伪代码:
if(队列为空 || 队尾值 > X)
插入队尾
2 X>=q[tail]
此时,队列递减性被打破,此时我们做一下操作:
① 弹出队尾元素
因为当前的 q[tail] 不但不是最大值,对于以后的情况也不如 X 更优,所以要弹出
②重复执行①,直到满足 X<q[tail] 或者队列为空为止
③将 X 插入队列
伪代码:
while(队列不为空 && 队尾的值 <= x)
从对尾弹出
队尾插入 x
询问的时候查询队头即可。
单调栈
这里不讲解单调栈的原理,只给出两种代码。
求数组中一个数左边第一个比它小的数的位置:
// 原数组为 a,下标从 1~n
// st 是 stack<int>
// 结果保存在 res,res[i] 表示 i 位置左边第一个比它小的数的下标
for(int i = 1;i <= n;++ i) {
while(!st.empty() && a[st.top()] >= a[i])
st.pop();
res[i] = st.empty() ? 0 : st.top();
st.push(i);
}
求数组中一个数右边第一个比它小的数的位置:
// 原数组为 a,下标从 1~n
// st 是 stack<int>
// 结果保存在 res,res[i] 表示 i 位置右边第一个比它小的数的下标
for (int i = 1; i <= n; ++i) {
while (!st.empty() && a[st.top()] > a[i]) {
res[st.top()] = i;
st.pop();
}
st.push(i);
}
while(!st.empty()) res[st.top()] = n + 1, st.pop();