scu 3139 单调队列

来源:互联网 发布:linux dns servfail 编辑:程序博客网 时间:2024/06/07 02:08

链接:http://acm.scu.edu.cn/soj/problem.action?id=3139
单调队列经典应用。
题意:有一个滑窗,长度为n,将滑窗一直往右边移动,确定每一次这个滑窗里面的最大值最小值是多少。
单调队列:
又称双端队列。以保存最大值为例。
队列中的元素是单调的不减的。
队头存放最大值,每次要将元素加入队列的时候,从队尾开始将比这个元素小的弹出,最后将这个元素加入队列。
每次判断队头的元素是否已经在滑窗外面,如果在的话就弹出。队列中每一个元素可以保存他在原数组中的索引和他自身的值。方便判断是否在滑窗内。
当然也可以只保存索引,然后去原数组中寻找就可以了。

#include <iostream>#include <cstdio>#include <cstring>#include <algorithm>using namespace std;#define M 1000009typedef struct{    int index;    int num;}node;node que1[M],que2[M]; //1 存放最小,2 存放最大int ans1[M],ans2[M];int main(){    //freopen("ans.txt","r",stdin);    int n,k;    while(scanf("%d %d",&n,&k) == 2)    {        int head1 = 0,tail1 = 0;        int head2 = 0,tail2 = 0;        for(int i = 0;i < n;i++)        {            int a;            scanf("%d",&a);            while(head1 != tail1 && que1[tail1-1].num > a) tail1--; //如果队列是空或者队尾元素大于,则减少尾指针            que1[tail1++] = {i,a};            if(que1[head1].index <= i-k) head1++;            while(head2 != tail2 && que2[tail2-1].num < a) tail2--;            que2[tail2++] = {i,a};            if(que2[head2].index <= i-k) head2++;            if(i >= k-1)            {                ans1[i-k+1] = que1[head1].num;                ans2[i-k+1] = que2[head2].num;            }        }        for(int i = 0;i <= n-k;i++) printf("%d%c",ans1[i],i==n-k ? '\n':' ');        for(int i = 0;i <= n-k;i++) printf("%d%c",ans2[i],i==n-k ? '\n':' ');    }    return 0;}
0 0
原创粉丝点击