滑动窗口_单调队列入门 (白书P241)

来源:互联网 发布:占领中环 知乎 编辑:程序博客网 时间:2024/05/17 03:40
 #include <iostream>#include <fstream>using namespace std;#define MAX 1000001int A[MAX]; //存储数据int Q[MAX]; //队列int P[MAX]; //存储A[i]中的下标iint 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;}

0 0
原创粉丝点击