栈
模板:
// tt表示栈顶
int stk[N], tt = 0;
// 向栈顶插入一个数
stk[ ++ tt] = x;
// 从栈顶弹出一个数
tt -- ;
// 栈顶的值
stk[tt];
// 判断栈是否为空,如果 tt > 0,则表示不为空
if (tt > 0)
{
}
队列
普通队列
模板:
// hh 表示队头,tt表示队尾
int q[N], hh = 0, tt = -1;
// 向队尾插入一个数
q[ ++ tt] = x;
// 从队头弹出一个数
hh ++ ;
// 队头的值
q[hh];
// 判断队列是否为空,如果 hh <= tt,则表示不为空
if (hh <= tt)
{
}
循环队列
模板:
// hh 表示队头,tt表示队尾的后一个位置
int q[N], hh = 0, tt = 0;
// 向队尾插入一个数
q[tt ++ ] = x;
if (tt == N) tt = 0;
// 从队头弹出一个数
hh ++ ;
if (hh == N) hh = 0;
// 队头的值
q[hh];
// 判断队列是否为空,如果hh != tt,则表示不为空
if (hh != tt)
{
}
单调栈
作用:给定一个序列,找出每个数左边离它最近的比它大/小的数
模板:
常见模型:找出每个数左边离它最近的比它大/小的数
int tt = 0;
for (int i = 1; i <= n; i ++ )
{
while (tt && check(stk[tt], i)) tt -- ;
stk[ ++ tt] = i;
}
原理:
- 遍历数组元素,每处理完一个元素就将它放入栈中,根据后进先出的特性,栈顶元素就是当前元素左边的最近元素,从栈顶开始往外弹出元素直到找到一个比当前元素小的元素停止。问题在于我们弹出的元素是否可能是接下来元素的答案?如果下一个元素大于当前元素,那么当前元素就是下一个元素的答案,与我们弹出的元素无关;如果下一个元素小于等于当前元素,因为弹出的元素都是大于等于当前元素的,所以弹出的元素依然不会是答案;说明弹出的元素不会影响后续结果,对每个元素进行入栈或出栈操作只可能为一次,这也是为什么这个问题可以用单调栈解决。
测试:
/*
题目描述:
给定一个长度为N的整数数列,输出每个数左边第一个比它小的数,如果不存在则输出-1
输入格式:
第一行包含整数N,表示数列长度
第二行包含N个整数,表示整数数列
输出格式:
1<=N<=10e5
1<=元素值<=10e9
输入样例:
5
3 4 2 7 5
输出样例:
-1 3 -1 2 2
*/
#include <iostream>
using namespace std;
const int N = 10e5 + 10;
int n;
int stk[N], tt;
int main(){
scanf("%d", &n);
for(int i = 0; i < n; i ++){
int x;
scanf("%d",&x);
while(tt && stk[tt] >= x) tt --;
if(tt) printf("%d ", stk[tt]);
else printf("-1 ");
stk[ ++ tt] = x;
}
return 0;
}
单调队列
作用:求滑动窗口中的最大值/最小值
模板:
常见模型:找出滑动窗口中的最大值/最小值
int hh = 0, tt = -1;
for (int i = 0; i < n; i ++ )
{
while (hh <= tt && check_out(q[hh])) hh ++ ; // 判断队头是否滑出窗口
while (hh <= tt && check(q[tt], i)) tt -- ;
q[ ++ tt] = i;
}
原理:
- 以 求滑动窗口中的最小值 为例,
- 窗口位置 最小值
1 [3 -1 -3] 5 3 6 7 -3 - 当下队列出队了 1,入队了 -3 。3 在 -3 的左边,之后 3 肯定比 -3 先弹出去且 3 比 -3 大,换言之只要 -3 在一天,3 就永远不可能作为最小值输出。-1 也是同理。所以只要有元素 在当前入队元素左边且比入队元素大,就把该元素删除,剩下的队列元素就是单调递增的。这个递增序列的第一个元素就是最小值。
- 怎么知道队首元素,什么时候出队?q[ hh ] 队列存的是下标,不是元素值。我们知道当前元素的下标 i 与窗口长度 k ,判断队头下标 q[ hh ] 不在 i - k +1 ~ i 之间,则队头元素 a[ q[ hh ] ] 出队。
- 求最大值同理。
测试:
/*
题目描述:
给定一个大小为 n<=10e6 的数组
有一个大小为 k 的滑动窗口,它从数组的最左边移动到最右边
你只能在窗口中看到k个数字
每次滑动窗口向右移动一个位置。
以下是一个例子:
该数组为[1 3 -1 -3 5 3 6 7],k为3
窗口位置 最小值 最大值
[1 3 -1] -3 5 3 6 7 -1 -3
1 [3 -1 -3] 5 3 6 7 -3 3
1 3 [-1 -3 5] 3 6 7 -3 5
1 3 -1 [-3 5 3] 6 7 -3 5
1 3 -1 -3 [5 3 6] 7 3 6
1 3 -1 -3 5 [3 6 7] 3 7
您的任务是确定滑动窗口位于每个位置时,窗口中的最大值和最小值
输入格式
第一行包含两个整数n和k,分别代表数组长度和滑动窗口的长度
第二行有n个整数,代表数组的具体数值
同行数据之间用空格隔开
输出格式
第一行输出,从左至右,每个位置滑动窗口中的最小值
第二行输出,从左至右,每个位置滑动窗口中的最大值
输入样例:
8 3
1 3 -1 -3 5 3 6 7
输出样例:
-1 -3 -3 -3 3 3
3 3 5 5 6 7
*/
#include <iostream>
using namespace std;
const int N = 1e6 + 10;
int n, k;
int a[N], q[N];
int main(){
scanf("%d%d", &n, &k);
for(int i = 0; i < n; i ++) scanf("%d", &a[i]);
int hh = 0, tt = -1;
for(int i = 0; i < n; i ++){
//判断对头是否已经滑出窗口
if(hh <= tt && i - k + 1 > q[hh]) hh ++;
while(hh <= tt && a[q[tt]] >= a[i]) tt --;
q[ ++ tt] = i;
if(i >= k - 1) printf("%d ", a[q[hh]]);
}
puts("");
hh = 0, tt = -1;
for(int i = 0; i < n; i ++){
//判断对头是否已经滑出窗口
if(hh <= tt && i - k + 1 > q[hh]) hh++;
while(hh <= tt && a[q[tt]] <= a[i]) tt --;
q[ ++ tt] = i;
if(i >= k - 1) printf("%d ", a[q[hh]]);
}
puts("");
return 0;
}