Poj 2823 Sliding Window(单调队列)

来源:互联网 发布:韩国聊天软件talk 编辑:程序博客网 时间:2024/06/15 22:09
题目地址:http://poj.org/problem?id=2823

思路:设置两单调队列:q(存储从小到大的值的下标)、Q(存储从大到小的值的下标)(初始时,两队列仅一元素1)。从1--->n依次扫描数组中元素:对于q,若当前值大于或等于队尾元素则将其下标i入队。否则,不断弹出队尾元素直到当前值不小于队尾元素x对应的a[x],若当前窗口元素个数大于k(i-队头元素>=k),则不断出队,直到i-队头元素<k(队列存储下标的作用);同理,对于Q,若当前值小于或等于队尾元素,则将其下标i入队。否则,不断弹出队尾元素直到q[tail]>=a[i],当窗口元素大于k时,不断从队头出队。当i>=k时q、Q队头元素的值即为最小值与最大值元素的下标,记录即可。

#include<cstdio>#include<cstring>#include<iostream>#include<algorithm>using namespace std;const int INF=0x3f3f3f3f;const int maxn=1e6+50;int n,k;int minans[maxn];int maxans[maxn];int head1,tail1,head2,tail2;int a[maxn],q[maxn],Q[maxn];int main(){    scanf("%d%d",&n,&k);    for(int i=1; i<=n; i++) scanf("%d",&a[i]);    head1=0,tail1=1,head2=0,tail2=1;    q[tail1]=1,Q[tail2]=1;    for(int i=1; i<=n; i++)    {        while(a[i]<a[q[tail1]]&&head1<=tail1) tail1--;        if(i!=1) tail1++,q[tail1]=i;        while(i-q[head1]>=k&&head1<=tail1) head1++;        if(i>=k) minans[i]=a[q[head1]];        while(a[i]>a[Q[tail2]]&&head2<=tail2) tail2--;        if(i!=1) tail2++,Q[tail2]=i;        while(i-Q[head2]>=k&&head2<=tail2) head2++;        if(i>=k) maxans[i]=a[Q[head2]];    }    for(int i=k; i<=n; i++)    {        if(i==k) printf("%d",minans[i]);        else printf(" %d",minans[i]);    }    printf("\n");    for(int i=k; i<=n; i++)    {        if(i==k) printf("%d",maxans[i]);        else printf(" %d",maxans[i]);    }    printf("\n");    return 0;}



0 0
原创粉丝点击