【双端队列】滑动的窗户 window

滑动的窗户

window.pas/c/cpp

3S/256MB

【题目描述】

       在一个包含n个元素的数组上，有一个长度为k的窗户在从左向右滑动。窗户每滑动到一个位置，我们都可以看到k个元素在窗户中。如下的例子所示，假设数组为 [1 3 -1 -3 5 3 6 7]，而k等于3:

 

窗户位置	最小值	最大值
[1  3  -1] -3  5  3  6  7	-1	3
1 [3  -1  -3] 5  3  6  7	-3	3
1  3 [-1  -3  5] 3  6  7	-3	5
1  3  -1 [-3  5  3] 6  7	-3	5
1  3  -1  -3 [5  3  6] 7	3	6
1  3  -1  -3  5 [3  6  7]	3	7

 

       对于窗户滑动过的每个位置，请给出窗户内k个元素的最小值和最大值。

【输入】

输入文件：window.in

输入的第一行包括两个整数n，k，n表示数组的长度，k表示窗户的长度。

接下来一行包括n个整数，表示这个n个元素的数组。

【输出】

输出文件：window.out

输出包含两行，每行包括n-k+1个整数，第一行表示窗户从左到右滑动过程中的最小值，第二行表示窗户从左到右滑动过程中的最大值。

【输入样例】

8 3

1 3 -1 -3 5 3 6 7

【输出样例】

-1 -3 -3 -3 33

3 3 5 5 6 7

【数据范围】

     对于100%的数据，3<=n<=1000000，1<=k<=n，数组中的每个元素均在int范围内

 

O(n3)的算法很容易想到，然后枚举窗户左界l，和窗户右界r，然后又从l到r枚举找出最值即可

但是看看题目数据范围，肯定会T！

 

我们以样例找最小值为例

13 -1 -3 1 3 6 7，窗户长度为3，当窗户在最开始时（即[1  3  -1] -3  1  3  6  7），我们发现因为有-1的存在，那么1和3都永远没用了！所以我们根本可以不看1和3，然后窗口继续向后滑，1 [3  -1  -3] 1  3  6  7，同理3和-1也就没用了，继续向后，1  3 [-1  -3  1] 3  6  7，那么-1也就没用了，而1比-3大，是不是也不要了呢？肯定还是要的！当窗口滑到如下时：1 3  -1  -3 [1  3  6]  7，1就派上了用场！

所以我们可以得到一个结论：只要有一个当前最小值的存在，那么它前面比它大的值肯定是没有用的！因为只要它前面那些值在窗户里，它肯定也在窗户里，而最小值又肯定是它！比如当窗户在如下位置时1 [3  -1  -3] 1  3  6  7，因为有了-3的存在，那么3和-1就肯定可以不要了！

基于上面的结论，我们就可以用一个队列来维护当前的值，并保证队列是单调递增的（那么队列第一个值肯定就是最小值！），那么我们怎样保证单调递增呢？我们每扫到一个数x，就把当前数和队列中最后一个数y（队列中最大的数）进行比较，如果x>y就把x如队（前面已经分析过了虽然是找最小值但也不能把较大的x直接扔掉的原因），否则我们就应该把队尾的数扔掉

这里问题就来了，我们的队列称作FIFO，全称是firstin, first out ，顾名思义“先进先出”，说白了，也就是从队尾入队，从队头出队，但是经前面的分析，我们要把队尾的数扔掉，也就是要从队尾出队，这样两端都可以出队的队列我们称作“双端队列”。在C++中我们自然可以手动维护队列，当然也可以用和queue类似的deque（自己百度一下吧。。。。）

维护队列单调性和队列存储的问题解决了，问题基本上就解决了。

不过还有一个问题！当窗户滑到1  3  -1  -3 [5  3  6] 7时，我们队列中的-3就应该出队了，那么我们该如何判断？扫一遍？这样效率又慢了。。。。。入队的时候，以前我们是把a[i]那个值入队，那么我们换一种思路，我们如果把i入队的话，我们就可以很方便的判断当前点i是否在窗户之外了。

 

C++ Code

/*
C++ Code
http://blog.csdn.net/jiangzh7
By Jiangzh
*/
#include<cstdio>
#include<algorithm>
using namespace std;

const int MAXN=1000000+10;

int n,k;
int a[MAXN];
int q[MAXN],head,tail;

void read()
{
	freopen("window.in","r",stdin);
	freopen("window.out","w",stdout);
	scanf("%d%d",&n,&k);
	for(int i=1;i<=n;i++) scanf("%d",&a[i]);
}

void work()
{
	//找最小值  区间严格单调递增
	memset(q,0,sizeof(q));head=tail=0;
	for(int i=1;i<=n;i++)
	{
		while(tail>head && a[i]<=a[q[tail-1]]) tail--;
		q[tail++]=i;
		if(q[head]+k-1<i) head++;
		if(i>=k) printf("%d ",a[q[head]]);
	}
	printf("\n");
	//找最大值  区间严格单调递减
	memset(q,0,sizeof(q));head=tail=0;
	for(int i=1;i<=n;i++)
	{
		while(tail>head && a[i]>=a[q[tail-1]]) tail--;
		q[tail++]=i;
		if(q[head]+k-1<i) head++;
		if(i>=k) printf("%d ",a[q[head]]);
	}
}

int main()
{
	read();
	work();
	return 0;
}