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

原题和样例见洛谷：滑动窗口

分析

这题是单调队列的模板题

队列就是先进先出的数据结构。队列的进出方式类似于我们排队打饭，来排队的人从队尾进入，打完饭的人从前面出来。有头指针和尾指针控制。
队列的单调性就是一个队列保持递增或者递减。

举个例子：假如你在饭堂打饭时，有个人人高马大，急匆匆跑过来，看排了这么一长串队，心中急躁，从队列最后的一个人开始，看见好欺负的就赶走，自己站着，直到干不过的就停下，这个人的插队行为类似于下面这幅图。

图中，企图插队的人战斗力为6，队尾的5，4，1都小于它，,6把这三个人全部挤掉，到了7的后面。 这就是所谓的单调队列了，队列元素保持单调递增（减），而保持的方式就是通过插队，把队尾破坏了单调性的数全部挤掉，从而使队列元素保持单调。

这就是单调队列的维护。

说回这道题

这题本身是可以双端弹出，一端压入，因为前面的如果过了“窗口”就没有用了。
还有，单调队列的最小（最大）值一定在队首，因为递减/递增单调性。

程序实现过程中先将前k个元素入队，此后每次在队尾加入a[k+1…n]，在插入元素中同时进行以下操作：

1. 将队尾所有大于a[i]的值弹出队列
2. 插入a[i]到队尾
3. 判断队首元素位置是否超出i−k，如果超出，那这个数就没用了，从队头弹出。

建立两个单调队列，一个递增，一个递减，最后取队首存起来输出。

上代码

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
int n,k,h,t,q[1000000],a[1000000]; 
int main()
{
    cin>>n>>k;
    for(int i=1;i<=n;i++)
    {
    	cin>>a[i];
    }
    h=0;t=1;
    for(int i=1;i<=n;i++)//最小 
    {
    	while(h<=t&&q[h]<i-k+1)
    	{
    		h++;//从头出队 
    	}
    	while(h<=t&&a[i]<a[q[t]])
		{
			t--;//比这个小的也被干掉 
		} 
    	t++;
    	q[t]=i;
    	if(i>=k)
		{
			cout<<a[q[h]]<<' ';//最优的永远在最前面 
		} 
    }
    cout<<endl;
	h=0;t=1;
	for(int i=1;i<=n;i++)
	{
		while(h<=t&&q[h]<i-k+1)//最大 
		{
			h++;
		}
		while(h<=t&&a[i]>a[q[t]])
		{
			t--;
		}
		t++;
		q[t]=i;
		if(i>=k)
		{
			cout<<a[q[h]]<<' ';
		}
	}
	return 0;
}

Update on 2022.1.6

另一种写法：

#include<iostream>
#include<cstdio>
#include<queue>
#include<algorithm>
using namespace std;

int n,a[1000010],k;

deque<int> q;

int main()
{
	scanf("%d%d",&n,&k);
	for(register int i=1;i<=n;i++)
	{
		scanf("%d",&a[i]);
	}
	for(register int i=1;i<=n;i++)
	{
		while(!q.empty()&&i-q.front()>=k) q.pop_front();
		while(!q.empty()&&a[i]<=a[q.back()]) q.pop_back();
		q.push_back(i);
		if(i>=k) cout<<a[q.front()]<<' ';
	}
	printf("\n");
	while(!q.empty()) q.pop_back();
	for(register int i=1;i<=n;i++)
	{
		while(!q.empty()&&i-q.front()>=k) q.pop_front();
		while(!q.empty()&&a[i]>=a[q.back()]) q.pop_back();
		q.push_back(i);
		if(i>=k) cout<<a[q.front()]<<' ';
	}
	return 0;
}