poj-2823 Sliding Window

最新推荐文章于 2019-07-23 09:58:49 发布

原创最新推荐文章于 2019-07-23 09:58:49 发布 · 657 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#栈 #数据结构

数据结构专栏收录该内容

19 篇文章

订阅专栏

本文介绍了一种使用单调队列解决区间最大值问题的方法，通过维护两个单调队列来高效地求解每个区间的最大值。

http://poj.org/problem?id=2823

基础的单调队列，维护一个单调递增和一个单调递减队列即可。

#include<stdio.h>
#define M 2000005
int a[M],Q[M],I[M];
int i,n,k;
void getMax() //求每个区间的最大值 维护一个单调递减的队列，则队首元素就是这个区间的最大值，如果不在范围内就队首元素加1
{
    int head=1,tail=0; //队首指针   队尾指针
    for(i=1;i<k;i++) //预处理第一个区间
	{
        while(head<=tail&&Q[tail]<=a[i])tail--;  //维护一个单调递减序列，如果队列不空并且当前元素大于等于队尾元素 队尾元素弹出，
        tail++;   //在把当前元素赋值队尾元素
        Q[tail]=a[i];I[tail]=i;  //并且记录下标
    }
    for(i=k;i<=n;i++)
	{
        while(head<=tail&&Q[tail]<=a[i])tail--;
        tail++;
        Q[tail]=a[i];I[tail]=i;
        while(I[head]<=i-k)head++; //判断当前队首元素是否在在范围内
        printf("%d ",Q[head]); //然后输出即可
    }
}
void getMin(){  //同理  维护一个单调递增的队列
    int head=1,tail=0;
    for(i=1;i<k;i++)
	{
        while(head<=tail&&Q[tail]>=a[i])tail--;
        tail++;
        Q[tail]=a[i];I[tail]=i;
    }
    for(i=k;i<=n;i++)
	{
        while(head<=tail&&Q[tail]>=a[i])tail--;
        tail++;
        Q[tail]=a[i];I[tail]=i;
        while(I[head]<=i-k) head++;
        printf("%d ",Q[head]);
    }
}
int main()
{
    while(scanf("%d %d",&n,&k)!=EOF)
    {
    for(i=1;i<=n;i++) scanf("%d",&a[i]);
    getMin();
    printf("\n");
    getMax();
    }
    return 0;
}