算法基础—数据结构—栈与队列

栈——先进后出；

队列——先进先出；

这两个模型应该都不会陌生，并且有相关的头文件可以直接调用，但是，这里依旧是拿数组进行模拟；

先来看栈的相关模板（取自AcWing）

// tt表示栈顶
int stk[N], tt = 0;

// 向栈顶插入一个数
stk[ ++ tt] = x;

// 从栈顶弹出一个数
tt -- ;

// 栈顶的值
stk[tt];

// 判断栈是否为空
if (tt > 0)
{

}

之后是队列的模板

普通队列

// hh 表示队头，tt表示队尾
int q[N], hh = 0, tt = -1;

// 向队尾插入一个数
q[ ++ tt] = x;

// 从队头弹出一个数
hh ++ ;

// 队头的值
q[hh];

// 判断队列是否为空
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];

// 判断队列是否为空
if (hh != tt)
{

}

之后就是单调栈和单调队列了

先是单调栈的模板

常见模型：找出每个数左边离它最近的比它大/小的数
int tt = 0;
for (int i = 1; i <= n; i ++ )
{
    while (tt && check(stk[tt], i)) tt -- ;
    stk[ ++ tt] = i;
}

之后是单调队列的模板

常见模型：找出滑动窗口中的最大值/最小值
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;
}

基本上栈和队列大部分的算法都在这里了，之后我们来看一些题目吧

首先是模拟栈的

https://www.acwing.com/problem/content/830/

套模板即可；

#include<bits/stdc++.h>
using namespace std;

const int N = 1e5+9;
int stk[N],tt = 0;

int main(){
    int n;
    cin >> n;
    while(n--){
        string op;
        int x;
        cin >> op;
        if(op=="push"){
            cin >> x;
            stk[++tt] = x;
        }
        else if(op=="query"){
            printf("%d\n",stk[tt]);
        }
        else if(op=="pop"){
            tt--;
        }
        else{
            if(tt>0) printf("NO\n");
            else printf("YES\n");
        }
    }
}

接下来是模拟队列

https://www.acwing.com/problem/content/831/

套用第一个普通队列的模板即可

#include<bits/stdc++.h>
using namespace std;

const int N = 1e5+9;
int hh = 0,tt = -1,stk[N];

int main(){
    int n;
    cin >> n;
    while(n--){
        string op;
        cin >> op;
        int x;
        if(op=="push"){
            cin >> x;
            stk[++tt] = x;
        }
        else if(op=="pop"){
            hh++;
        }
        else if(op=="query"){
            printf("%d\n",stk[hh]);
        }
        else if(op=="empty"){
            if(hh>tt) printf("YES\n");
            else printf("NO\n");
        }
    }
}

之后是单调栈的题目；

https://www.acwing.com/problem/content/832/

其实大体思路就是如果栈顶的值大则出栈，直到有较小的值为止，输出栈顶，如果栈为空则为-1.

#include<bits/stdc++.h>
using namespace std;

const int N = 1e5+9;
int stk[N],tt;

int main(){
    int n;
    cin >> n;
    for(int i=0;i<n;i++){
        int x;
        cin >> x;
        while(tt&&stk[tt] >=x) tt--;
        if(tt) cout << stk[tt] << " ";
        else cout << "-1 ";
        
        stk[++tt] = x;
    }
    return 0;
}

最后一个就是单调队列了

https://www.acwing.com/problem/content/156/

这里队列存的是下标，来看是否超出滑动的长度，队列里的下标所对应的元素是有单调性的，看新加入的元素是否比队尾下标对应的元素小，如果小，则插入到相应的位置。

#include<bits/stdc++.h>
using namespace std;

const int N = 1e6+9;
int stk[N],q[N],m,n;

int main(){
    cin >> n >> m;
    for(int i=0;i<n;i++){
        scanf("%d",&q[i]);
    }
    int tt = -1,hh = 0;
    for(int i=0;i<n;i++){
        while(hh<=tt&&stk[hh]<i-m+1) hh++;
        while(hh<=tt&&q[stk[tt]]>=q[i]) tt--;
        stk[++tt] = i;
        if(i>=m-1) printf("%d ",q[stk[hh]]);
    }
    printf("\n");
    tt = -1,hh = 0;
    for(int i=0;i<n;i++){
        while(hh<=tt&&stk[hh]<i-m+1) hh++;
        while(hh<=tt&&q[stk[tt]]<=q[i]) tt--;
        stk[++tt] = i;
        if(i>=m-1) printf("%d ",q[stk[hh]]);
    }
    printf("\n");
}