码迷,mamicode.com
首页 > 其他好文 > 详细

luogu P1886 滑动窗口(单调队列

时间:2017-10-15 22:27:40      阅读:164      评论:0      收藏:0      [点我收藏+]

标签:++i   --   str   include   print   const   记录   code   数据   

题目描述

现在有一堆数字共N个数字(N<=10^6),以及一个大小为k的窗口。现在这个从左边开始向右滑动,每次滑动一个单位,求出每次滑动后窗口中的最大值和最小值。

例如:

The array is [1 3 -1 -3 5 3 6 7], and k = 3.

技术分享

输入输出格式

输入格式:

 

输入一共有两行,第一行为n,k。

第二行为n个数(<INT_MAX).

 

输出格式:

 

输出共两行,第一行为每次窗口滑动的最小值

第二行为每次窗口滑动的最大值

 

输入输出样例

输入样例#1:
8 3
1 3 -1 -3 5 3 6 7
输出样例#1:
-1 -3 -3 -3 3 3
3 3 5 5 6 7

说明

50%的数据,n<=10^5

100%的数据,n<=10^6

单调队列版,单调队列内记录的为下标

#include<cstdio>
#include<iostream>
#include<algorithm>
using namespace std;
const int maxn = 1000003;
int n,k;
int a[maxn];
int q1[maxn],q2[maxn];
int mx[maxn],mn[maxn];
int main() {
    scanf("%d%d",&n,&k);
    for(int i=1;i<=n;++i) scanf("%d",a+i);
    int h1,h2,t1,t2;
    q1[1]=q2[1]=1;
    h1=h2=t1=t2=1;int cnt=0;//5 1 1 2 3 4 5
    if(k==1) {
        for(int i=1;i<=n;i++)cout<<a[i]<<" ";
        cout<<endl;
        for(int i=1;i<=n;i++)cout<<a[i]<<" ";return 0;
    }
    for(int i=2;i<=k;++i) {
        while(h1<=t1&&a[q1[t1]]<a[i])t1--;
        while(h2<=t2&&a[q2[t2]]>a[i])t2--;
        q1[++t1]=q2[++t2]=i;
    }
    mx[++cnt]=a[q1[h1]],mn[cnt]=a[q2[h2]];
    for(int i=k+1;i<=n;++i) {
        while(q1[h1]<i-k+1)
        h1++;
        while(q2[h2]<i-k+1)
        h2++;
        while(h1<=t1&&a[q1[t1]]<a[i])t1--;
        while(h2<=t2&&a[q2[t2]]>a[i])t2--;
        q1[++t1]=q2[++t2]=i;
        mx[++cnt]=a[q1[h1]];mn[cnt]=a[q2[h2]];
    }
    for(int i=1;i<=cnt;++i) 
        printf("%d ",mn[i]);
    puts("");
    for(int i=1;i<=cnt;++i)
        printf("%d ",mx[i]);
    return 0;
}

 

luogu P1886 滑动窗口(单调队列

标签:++i   --   str   include   print   const   记录   code   数据   

原文地址:http://www.cnblogs.com/sssy/p/7674320.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!