Week5 HomeWork D 滑动窗口 单调队列

题目描述

ZJM 有一个长度为 n 的数列和一个大小为 k 的窗口, 窗口可以在数列上来回移动. 现在 ZJM 想知道在窗口从左往右滑的时候，每次窗口内数的最大值和最小值分别是多少. 例如：
数列是 [1 3 -1 -3 5 3 6 7], 其中 k 等于 3.

Window position	Minimum value	Maximum value
[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

Input

输入有两行。第一行两个整数n和k分别表示数列的长度和滑动窗口的大小，1<=k<=n<=1000000。第二行有n个整数表示ZJM的数列。

Output

输出有两行。第一行输出滑动窗口在从左到右的每个位置时，滑动窗口中的最小值。第二行是最大值。

Sample Input

8 3
1 3 -1 -3 5 3 6 7

Sample Output

-1 -3 -3 -3 3 3
3 3 5 5 6 7

算法/思路分析

本题要求局部的最值，维护单调队列记录最值比较方便。具体而言：维护一个单调递增队列，队列中的元素均属于当前窗口，非当前窗口的队首元素弹出，当前窗口的元素均入队后，队首即为当前窗口的最小值。同理：维护一个单调递减队列，队首即为当前窗口的最大值。

注意：无需判断队列内每个元素是否属于当前窗口，只需判断队首即可。因为，每次移动窗口，只有队首元素有可能不属于当前窗口。

代码

#include <cstdio>
#include <iostream>
#include <stack>
#include <algorithm>
#include <cmath>
#include <deque>
using namespace std;
const int maxn = 1e6+10;
#define ll long long
int n, k ,a[maxn],M[maxn] = {0},m[maxn] = {0};

/*
bool belong(int p, int l, int r)
{
	bool type = false;
	for(int i = l;i <= r; i++)
	{
		if(p == a[i]) 
		{
			type = true;
			break;
		}
	}
	return type;
}*/

int main()
{
	//q1为单调递增队列，便于记录最小值
	//q2为单调递减队列，便于记录最大值 
	deque<int> q1,q2;
	q1.clear();
	q2.clear(); 
	
	scanf("%d%d",&n,&k);
	for(int i = 0; i < n; i++)	scanf("%d",&a[i]);
	q1.push_back(a[0]); q2.push_back(a[0]);	
	
	//第1个窗口 
	for(int i = 1; i <= k-1; i++)
	{
		while(q1.size() > 0 && q1.back() > a[i]) q1.pop_back(); 
		q1.push_back(a[i]);
		
		while(q2.size() > 0 && q2.back() < a[i]) q2.pop_back(); 
		q2.push_back(a[i]);
	}
	m[0] = q1.front();
	M[0] = q2.front();
	
	//后续窗口 
	for(int i = k; i < n; i++)
	{
		//非当前窗口的元素出队 
		if(q1.front() == a[i-k]) q1.pop_front();
		//while(q1.size() > 0 && !belong(q1.front(),i-(k-1),i-1)) q1.pop_front(); 
		
		while(q1.size() > 0 && q1.back() > a[i]) q1.pop_back(); 
		q1.push_back(a[i]);
		m[i-(k-1)] = q1.front();
		
		//非当前窗口的元素出队
		if(q2.front() == a[i-k]) q2.pop_front();
		//while(q2.size() > 0 && !belong(q2.front(),i-(k-1),i-1)) q2.pop_front();
		
		while(q2.size() > 0 && q2.back() < a[i]) q2.pop_back(); 
		q2.push_back(a[i]);
		M[i-(k-1)] = q2.front();
	}
	
	//输出最小值 
	for(int i = 0; i < n - (k-1);i++) 
	{
		if(i > 0) printf(" ");
		printf("%d",m[i]);
	}
	printf("\n");
	
	//输出最大值 
	for(int i = 0; i < n - (k-1);i++) 
	{
		if(i > 0) printf(" ");
		printf("%d",M[i]);
	}
	printf("\n");
	
	return 0;
} 
/*
8 3
1 3 -1 -3 5 3 6 7
8 1
1 3 -1 -3 5 3 6 7
8 2
1 3 -1 -3 5 3 6 7
*/