模拟队列--基础课

板子

模拟队列

829. 模拟队列

#include <iostream>

using namespace std;

const int N = 1e5 + 10;

int n;

int tt, hh;
int q[N];


void init()
{
    // 初始化队列
    hh = tt = 0;
}



int main()
{
    // 对队列进行初始化
    init();
    
    cin >> n;
    
    while(n -- )
    {
        string op;
        int x;
        
        cin >> op;
        
        if(op == "push")
        {
            cin >> x;
            // 队尾插入一个数
            q[tt ++] =x;
        } else if(op == "pop")
        {
            // 队头弹出一个数
            hh ++ ;
        } else if(op == "query")
        {
            // 查询对头元素
            cout << q[hh] << endl;
        } else {
            // 判断是否为空
            if(hh == tt) puts("YES");
            else puts("NO");
        }
    }
    
    return 0;
}

注意

• 模拟队列中，队尾在后面，队头靠近0
  
• 队尾加数向右走， 队头加数向右走
  

总结

// 初始化队列
hh = tt = 0;

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

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

// 查询对头元素
q[hh]
 
// 判断是否为空
hh == tt

单调队列

154. 滑动窗口

#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 tt = -1, hh = 0;
    for(int i = 0; i < n; i ++)
    {
        // 队列不空 &&  对头离开窗口 -- 正常移动
        if(hh <= tt && q[hh] < i - k  + 1 ) 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 && q[hh] < i - k + 1) hh ++ ;
        while(hh <= tt && a[q[tt]] <= a[i]) tt --;
        q[++ tt ] = i;
        
        if(i >= k - 1) printf("%d ", a[q[hh]]);
    }
    
    
    return 0;
}

总结

//主要是运用队列性质
//这里采取了双端队列：两边都可以入队，出队
//队列中保存元素下标：方便计算是否划出窗口

注意

// 本题中对队列初始化有所不同 
// 采取tt = -1 , hh = 0
// 与模拟队列相比，只有在判断队列是否为空的条件不同