单调队列滑动窗口——洛谷P1886

单调队列,故名思义,就是队列中的元素单调递增或者递减。
然后考察队首(head)和队尾(tail)
对于队尾,主要是让元素进入,如果进来的元素打乱了原先的顺序,那么就剔除队尾的一些元素,再加入新的元素,直到符合单调性。
对于队首,如果超出了滑动窗口的大小那么,就要弹出元素。
code

#include<bits/stdc++.h>
#define maxn 1000010
using namespace std;
int a[maxn],q[maxn],p[maxn],head,tail;int n,k;
void ask_min()
{
    
    
	head = 1;
	tail = 0;
	for(int i=1;i<=n;i++)
	{
    
    
		while(head<=tail&&q[tail]>=a[i])//队尾弹出元素
		{
    
    
			tail--;
		}
		q[++tail] = a[i];//加入当前的元素
		p[tail] = i;//记录元素下标
		while(p[head]<=i-k)//头元素的下标在窗口之外
		{
    
    
			head++;//弹出头部元素
		}
		if(i>=k)printf("%d ",q[head]);
	}
	printf("\n");
}
void ask_max()
{
    
    
	head = 1;
	tail = 0;
	for(int i=1;i<=n;i++)
	{
    
    
		while(head<=tail&&q[tail]<=a[i])
		{
    
    
			tail--;
		}
		q[++tail] = a[i];
		p[tail] = i;
		while(p[head]<=i-k)
		{
    
    
			head++;
		}
		if(i>=k)printf("%d ",q[head]);
	}
	printf("\n");
}
int main()
{
    
    
	
	cin>>n>>k;
	for(int i=1;i<=n;i++)
		scanf("%d",&a[i]);
	ask_min();ask_max();
}

猜你喜欢

转载自blog.csdn.net/naiue/article/details/107551345
今日推荐