poj 2838 Sliding Window

最新推荐文章于 2024-09-18 20:33:32 发布

HyogaHyoga

最新推荐文章于 2024-09-18 20:33:32 发布

阅读量743

点赞数

文章标签： mina 算法 im

本文链接：https://blog.csdn.net/HyogaHyoga/article/details/7829830

版权

题目大意：

给你n(<=10^6)个数和m，依次求出每个区间[i,i+m]的最大值与最小值。

题解：

n太大了，此题即使使用O（nlogm）的算法（比如线段树，RMQ等）的算法也有可能超时。考虑O（n）算法。

下面以求最大值为例：

设f[i]表示区间[i-m+1,i]的最大值，则有f[i]=max(a[j]),其中j满足b[i]<=j<i,其中b[i]=i-m+1;可见b[i]单调增加，满足单调队列条件。维护一个单调递减序列即可。

求最小值，则维护一个单调递增序列即可。

#include<iostream>
#include<cstdio>
#define inf (1<<30)
using namespace std;
int a[1000101],mina[1000101],maxa[1000101],q[1000101],tail,head;
int main()
{
    int n,m;
    while(scanf("%d%d",&n,&m)!=EOF)
    {
        for(int i=1;i<=n;i++)
        scanf("%d",&a[i]);
        head=1;tail=1;q[1]=1;
        for(int i=1;i<=n;i++)
        {
            while(head<=tail&&q[head]<i-m+1)head++;
            while(head<=tail&&a[i]>=a[q[tail]])tail--;
            tail++;q[tail]=i;
            maxa[i]=a[q[head]];
        }
        head=1;tail=1;q[1]=1;
        for(int i=1;i<=n;i++)
        {
            while(head<=tail&&q[head]<i-m+1)head++;
            while(head<=tail&&a[i]<=a[q[tail]])tail--;
            tail++;q[tail]=i;
            mina[i]=a[q[head]];
        }
        for(int i=m;i<=n-1;i++)
        printf("%d ",mina[i]);
        printf("%d\n",mina[n]);
        for(int i=m;i<=n-1;i++)
        printf("%d ",maxa[i]);
        printf("%d\n",maxa[n]);
    }
    return 0;
}