滑动窗口——单调队列

滑动窗口 - LibreOJ 10175 - Virtual Judge (vjudge.net)

给一个长度为 N 的数组，一个长为 K 的滑动窗体从最左端移至最右端，你只能看到窗口中的 K 个数，每次窗体向右移动一位，如下图：

窗口位置	最小值	最大值
[1 3 -1] -3 5 3 6 7[1 3 -1] -3 5 3 6 7	−1−1	33
1 [3 -1 -3] 5 3 6 7  1 [3 -1 -3] 5 3 6 7	−3−3	33
1 3 [-1 -3 5] 3 6 7  1 3 [-1 -3 5] 3 6 7	−3−3	55
1 3 -1 [-3 5 3] 6 7  1 3 -1 [-3 5 3] 6 7	−3−3	55
1 3 -1 -3 [5 3 6] 7  1 3 -1 -3 [5 3 6] 7	33	66
1 3 -1 -3 5 [3 6 7]  1 3 -1 -3 5 [3 6 7]	33	77

你的任务是找出窗体在各个位置时的最大值和最小值。

输入格式

第 1 行：两个整数 N 和 K；
第 2 行：N 个整数，表示数组的 N 个元素(≤2×109≤2×109)；

输出格式

第一行为滑动窗口从左向右移动到每个位置时的最小值，每个数之间用一个空格分开；
第二行为滑动窗口从左向右移动到每个位置时的最大值，每个数之间用一个空格分开。

分析：

利用双向队列模拟单调队列。大致思路为：将每一个可能成为最大（最小）值的数存入队列。比如说当求最大值时，当前值为3此时队尾值为1，那么1已经没有成为最大值的可能了所以去除。相反如果当前值比队尾小，那么他在未来有可能成为某一时刻的最大值所以存入队列。求最小值同理。

下面贴上代码：

#include<map>
#include<algorithm>
#include<stdio.h>
#include<string.h>
#include<deque>
#include<iostream>
using namespace std;
const int N=1e6+5,inf=0x3f3f3f3f;
int da[N],b[N],s[N];
int main()
{
	deque<int>p;
	deque<int>q;
	int n,k;
	scanf("%d %d",&n,&k);
	for(int i=1;i<=n;i++)
		scanf("%d",&da[i]);
	for(int i=1;i<=n;i++)//找最小
	{
		if(!p.empty()&&da[i]>=da[p.back()])
			p.push_back(i);
		else
		{
			while(!p.empty()&&da[i]<da[p.back()])
				p.pop_back();
			p.push_back(i);			
		}
		if(i>k&&((i-k)==p.front()))
			p.pop_front();
		if(i>=k)
			s[i]=da[p.front()];
	}	
	for(int i=1;i<=n;i++)//找最大
	{
		if(!q.empty()&&da[i]<=da[q.back()])
			q.push_back(i);
		else
		{
			while(!q.empty()&&da[i]>da[q.back()])
				q.pop_back();
			q.push_back(i);			
		}
		if(i>k&&((i-k)==q.front()))
			q.pop_front();
		if(i>=k)
			b[i]=da[q.front()];
	}
	for(int i=k;i<=n;i++)
		printf("%d ",s[i]);
	printf("\n");	
	for(int i=k;i<=n;i++)
		printf("%d ",b[i]);
	return 0;	
}