单调队列求第k大值

//单调队列求第k大值
#include <iostream>
#include <fstream>

using namespace std;

#define MAX 1000001
int A[MAX]; //存储数据
int Q[MAX]; //队列
int P[MAX]; //存储A[i]中的下标i
int Min[MAX];  //输出最小
int Max[MAX];  //输出最大
int n,k;

void get_min()
{
    int i;
    int head=1,tail=0;
    for(i=0; i<k-1; i++)  //先把前两个入队
    {
        while(head<=tail && Q[tail]>=A[i])  //队尾元素大于要插入的数
            --tail;
        Q[++tail]=A[i];
        P[tail]=i;
    }

    for(; i<n; i++)
    {
        while(head<=tail && Q[tail]>=A[i])
            --tail;
        Q[++tail]=A[i];
        P[tail]=i;
        while(P[head]<i-k+1)  //判断数是否过时，即窗口是否已经划过这个数，我这是从0开始计数的。
        {
            head++;
        }
        Min[i-k+1]=Q[head];
    }
}

void get_max()
{
    int i;
    int head=1,tail=0;
    for(i=0; i<k-1; i++)
    {
        while(head<=tail && Q[tail]<=A[i])   //队尾元素小于要插入的值
            --tail;
        Q[++tail]=A[i];
        P[tail]=i;
    }

    for(; i<n; i++)
    {

        while(head<=tail && Q[tail]<=A[i])   //队尾元素小于要插入的值
            --tail;
        Q[++tail]=A[i];
        P[tail]=i;
        while(P[head]<i-k+1)
            {
            head++;
        }
        Max[i-k+1]=Q[head];
    }
}

void output()
{
       int i;      
      //输出最下值 
       for(i=0; i<n-k+1; i++)
       {
           if(i==0)
               printf("%d",Min[i]);
           else
               printf(" %d",Min[i]);
       }
       printf("\n");
       //输出最大值
       for(i=0; i<n-k+1; i++)
       {
           if(i==0)
               printf("%d",Max[i]);
           else
               printf(" %d",Max[i]);
       }
       printf("\n");
}


int main()
{
    int i;
    freopen("acm.txt","r",stdin);
    scanf("%d%d",&n,&k);
    for(i=0; i<n; i++)
    {
        scanf("%d",&A[i]);
    }
    get_min();
    get_max();
    output();
    return 0;
}