week5作业题_D-滑动窗口

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
输出有两行。第一行输出滑动窗口在从左到右的每个位置时，滑动窗口中的最小值。第二行是最大值。
Exmaples
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

题目思路

这个题目需要我们做的是查找局部中的最小值和最大值，这样我们就可以维护一个单调队列，队列的元素均属于当前窗口。当元素不属于当前窗口时，需将队首元素弹出。这里我们为了代码简便使用的是deque结构，但是这样做耗时会长很多(这里就导致了我在G++环境下TLE)。首先初始化窗口，然后随着窗口的移动，将元素加入队列，如果保持单调性，就直接加入队尾，否则弹出队首元素，再将元素加入队尾。记录下每个窗口中的最小值或者是最大值。

代码实现

//#include <bits/stdc++.h>
#include <cstdio>
#include <iostream>
#include <deque>
using namespace std; 
#define _for(i,a,b) for(int i = (a);i < (b); i++)
#define _rep(i,a,b) for(int i = (a);i <= (b); i++)
const int N = 1e6 + 10;

int n,k;
int  Min[N],Max[N],a[N];
deque<int> p;		//索引单调队列 
deque<int> q;

void getMin()			//单调递增队列 
{
	_for(i,0,k - 1)			//初始化 
	{
		while(!p.empty() && a[p.back()] > a[i])
			p.pop_back();
		p.push_back(i);
	}
	_for(i,k - 1,n)
	{
		while(!p.empty() && a[p.back()] > a[i])
			p.pop_back();
		p.push_back(i);
		while(!p.empty() && (i - p.front() + 1) > k)
			p.pop_front();
		Min[i] = a[p.front()];
	}
	
}

void getMax()			//单调递减队列 
{
	_for(i,0,k - 1)		//初始化 
	{
		while(!q.empty() && a[q.back()] < a[i])
			q.pop_back();
		q.push_back(i);
	}
	_for(i,k - 1,n)
	{
		while(!q.empty() && a[q.back()] < a[i])
			q.pop_back();
		q.push_back(i);
		while(!q.empty() && (i - q.front() + 1) > k)
			q.pop_front();
		Max[i] = a[q.front()];
	}
	
}

int main()
{
	scanf("%d%d",&n,&k);
	_for(i,0,n)
		scanf("%d",&a[i]);
	p.clear();
	q.clear(); 
	getMin();
	getMax();
	
	_for(i,k - 1,n)
		printf("%d ",Min[i]);
	printf("\n");
	_for(i,k - 1,n)
		printf("%d ",Max[i]);
	printf("\n");
 } 

---

总结

1.这里需要注意到数据量很大，需要使用scanf和printf读入和输出，单纯用cin，cout会TLE。