P1886 滑动窗口 /【模板】单调队列

展开
题目描述
有一个长为 nn 的序列 aa,以及一个大小为 kk 的窗口。现在这个从左边开始向右滑动,每次滑动一个单位,求出每次滑动后窗口中的最大值和最小值。

例如:

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

在这里插入图片描述

输入格式
输入一共有两行,第一行有两个正整数 n,kn,k。 第二行 nn 个整数,表示序列 aa

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

输入输出样例
输入 #1复制
8 3
1 3 -1 -3 5 3 6 7
输出 #1复制
-1 -3 -3 -3 3 3
3 3 5 5 6 7
说明/提示

#include <cstdio>
#include <cstring>
#include <algorithm>

using namespace std;

const int maxn = 1e6 + 7;
int a[maxn];
int l1,r1,l2,r2;
int n,k;

struct QUE
{
    int i,x;
}q1[maxn],q2[maxn];

void solve1(int x)//递增
{
    while(l1 <= r1 && x - q1[l1].i >= k)
        l1++;
    while(l1 <= r1 && q1[r1].x > a[x])
        r1--;
    q1[++r1].x = a[x];q1[r1].i = x;
}

void solve2(int x)//递减
{
    while(l2 <= r2 && x - q2[l2].i >= k)
        l2++;
    while(l2 <= r2 && q2[r2].x < a[x])
        r2--;
    q2[++r2].x = a[x];q2[r2].i = x;
}

int main()
{
    scanf("%d%d",&n,&k);
    for(int i = 1;i <= n;i++)
    {
        scanf("%d",&a[i]);
    }
    
    for(int i = 1;i < k;i++)
    {
        solve1(i);solve2(i);
    }
    
    for(int i = k;i <= n;i++)
    {
        solve1(i);
        printf("%d ",q1[l1].x);
    }
    printf("\n");
    for(int i = k;i <= n;i++)
    {
        solve2(i);
        printf("%d ",q2[l2].x);
    }
    return 0;
}

发布了628 篇原创文章 · 获赞 17 · 访问量 2万+

猜你喜欢

转载自blog.csdn.net/tomjobs/article/details/104040703